Skip to content

Commit

Permalink
add some list size tests
Browse files Browse the repository at this point in the history
  • Loading branch information
coastalwhite committed Nov 21, 2024
1 parent fc928f7 commit d32c989
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions py-polars/tests/unit/test_row_encoding.py
Original file line number Diff line number Diff line change
Expand Up @@ -397,6 +397,13 @@ def test_order_list() -> None:
assert_order_series([None], [[1, 2, 3]], dtype, ["gt"], nulls_last=True)
assert_order_series([[None, 2, 3]], [[None, 2, 1]], dtype, ["gt"])

assert_order_series([[]], [[None]], dtype, ["lt"])
assert_order_series([[]], [[None]], dtype, ["lt"], descending=True)
assert_order_series([[]], [[1]], dtype, ["lt"])
assert_order_series([[]], [[1]], dtype, ["lt"], descending=True)
assert_order_series([[1]], [[1, 2]], dtype, ["lt"])
assert_order_series([[1]], [[1, 2]], dtype, ["lt"], descending=True)


def test_order_array() -> None:
dtype = pl.Array(pl.Int32, 3)
Expand Down

0 comments on commit d32c989

Please sign in to comment.