Update python/cudf/cudf/tests/test_csv.py

Co-authored-by: GALI PREM SAGAR <[email protected]>
rapidsai · Oct 31, 2022 · 99eedc0 · 99eedc0
1 parent 6710b47
commit 99eedc0
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/python/cudf/cudf/tests/test_csv.py b/python/cudf/cudf/tests/test_csv.py
@@ -789,9 +789,6 @@ def test_csv_reader_bools_custom():
         true_values=trues,
         false_values=falses,
     )
-    assert len(df.columns) == 2
-    assert df["text"].dtype == np.dtype("object")
-    assert df["int"].dtype == np.dtype("int64")
     expected = pd.read_csv(
         StringIO(buffer),
         names=names,
@@ -800,7 +797,7 @@ def test_csv_reader_bools_custom():
         true_values=trues,
         false_values=falses,
     )
-    assert_eq(df, expected)
+    assert_eq(df, expected, check_dtype=True)
 
 
 def test_csv_reader_bools_NA():