Skip to content

Commit

Permalink
Fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
staadecker committed May 22, 2024
1 parent 3872568 commit d995d65
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/pyoframe/io_mappers.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,7 +153,7 @@ def _to_base36(cls, int_col: pl.Series) -> pl.Series:
>>> import polars as pl
>>> s = pl.Series([0,10,20,60,53,66], dtype=pl.UInt32)
>>> Base36Mapper._to_base36(s).to_list()
['0', 'A', 'K', '1O', '1H', '1U']
['0', 'a', 'k', '1o', '1h', '1u']
>>> s = pl.Series([0], dtype=pl.UInt32)
>>> Base36Mapper._to_base36(s).to_list()
Expand Down Expand Up @@ -205,7 +205,7 @@ class Base36VarMapper(Base36Mapper):
>>> (m.x.filter(t=11)+1).to_str()
'[11]: 1 + x[11]'
>>> (m.x.filter(t=11)+1).to_str(var_map=Base36VarMapper(Variable))
'[11]: 1 + xB'
'[11]: 1 + xb'
>>> Base36VarMapper(Variable).apply(pl.DataFrame({VAR_KEY: []}))
shape: (0, 1)
Expand Down

0 comments on commit d995d65

Please sign in to comment.