From a3bf22992841d3ccc06c46ec3c72281966289742 Mon Sep 17 00:00:00 2001 From: Lawrence Mitchell Date: Fri, 12 Jul 2024 17:08:54 +0000 Subject: [PATCH] Add test --- python/cudf_polars/tests/test_hconcat.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/python/cudf_polars/tests/test_hconcat.py b/python/cudf_polars/tests/test_hconcat.py index 46cbb21b25a..4737aa18028 100644 --- a/python/cudf_polars/tests/test_hconcat.py +++ b/python/cudf_polars/tests/test_hconcat.py @@ -17,3 +17,12 @@ def test_hconcat(): ldf2 = ldf.select((pl.col("a") + pl.col("b")).alias("c")) query = pl.concat([ldf, ldf2], how="horizontal") assert_gpu_result_equal(query) + + +def test_hconcat_different_heights(): + left = pl.LazyFrame({"a": [1, 2, 3, 4]}) + + right = pl.LazyFrame({"b": [[1], [2]], "c": ["a", "bcde"]}) + + q = pl.concat([left, right], how="horizontal") + assert_gpu_result_equal(q)