Only wrap

rapidsai · Feb 5, 2022 · 0a80678 · 0a80678 · bdice · Feb 5, 2022
1 parent 1caa64f
commit 0a80678
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/python/cudf/cudf/tests/test_struct.py b/python/cudf/cudf/tests/test_struct.py
@@ -207,13 +207,11 @@ def test_dataframe_to_struct():
 
     # check that a non-string (but convertible to string) named column can be
     # converted to struct
+    df = cudf.DataFrame([[1, 2], [3, 4]], columns=[(1, "b"), 0])
+    expect = cudf.Series([{"(1, 'b')": 1, "0": 2}, {"(1, 'b')": 3, "0": 4}])
     with pytest.warns(UserWarning, match="will be casted"):
-        df = cudf.DataFrame([[1, 2], [3, 4]], columns=[(1, "b"), 0])
-        expect = cudf.Series(
-            [{"(1, 'b')": 1, "0": 2}, {"(1, 'b')": 3, "0": 4}]
-        )
         got = df.to_struct()
-        assert_eq(got, expect)
+    assert_eq(got, expect)
 
 
 @pytest.mark.parametrize(