add extra test for polars expand

pyjanitor-devs · Jul 19, 2024 · a46c1c8 · a46c1c8
1 parent 1e043d9
commit a46c1c8
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/polars/functions/test_expand_polars.py b/tests/polars/functions/test_expand_polars.py
@@ -43,3 +43,11 @@ def test_expand_1(df):
     """
     Test output for janitor.expand.
     """
+    expected = df.expand("group", "item_id", "item_name", sort=True)
+    actual = (
+        df.select(pl.col("group").unique())
+        .join(df.select(pl.col("item_id").unique()), how="cross")
+        .join(df.select(pl.col("item_name").unique()), how="cross")
+        .sort(by=pl.all())
+    )
+    assert_frame_equal(actual, expected)