Don't check dtype for scalar timestamps GPU tests

charlesbluca · charlesbluca · commit 0228514ceda1 · 2024-05-24T13:53:34.000-07:00
diff --git a/tests/integration/test_rex.py b/tests/integration/test_rex.py
@@ -1144,7 +1144,7 @@ def test_scalar_timestamps(c, gpu):
         }
     )
     df1 = c.sql("SELECT CEIL(to_timestamp(d) TO DAY) AS ceil_to_day FROM df")
-    assert_eq(df1, expected_df)
+    assert_eq(df1, expected_df, check_dtype=(not gpu))
     df2 = c.sql("SELECT CEIL(CAST(d AS TIMESTAMP) TO DAY) AS ceil_to_day FROM df")
     assert_eq(df2, expected_df)
 
@@ -1154,7 +1154,7 @@ def test_scalar_timestamps(c, gpu):
         }
     )
     df1 = c.sql("SELECT FLOOR(to_timestamp(d) TO DAY) AS floor_to_day FROM df")
-    assert_eq(df1, expected_df)
+    assert_eq(df1, expected_df, check_dtype=(not gpu))
     df2 = c.sql("SELECT FLOOR(CAST(d AS TIMESTAMP) TO DAY) AS floor_to_day FROM df")
     assert_eq(df2, expected_df)
 

Original file line number	Diff line number	Diff line change
`@@ -1144,7 +1144,7 @@ def test_scalar_timestamps(c, gpu):`
`1144`	`1144`	`}`
`1145`	`1145`	`)`
`1146`	`1146`	`df1 = c.sql("SELECT CEIL(to_timestamp(d) TO DAY) AS ceil_to_day FROM df")`
`1147`		`- assert_eq(df1, expected_df)`
	`1147`	`+ assert_eq(df1, expected_df, check_dtype=(not gpu))`
`1148`	`1148`	`df2 = c.sql("SELECT CEIL(CAST(d AS TIMESTAMP) TO DAY) AS ceil_to_day FROM df")`
`1149`	`1149`	`assert_eq(df2, expected_df)`
`1150`	`1150`
`@@ -1154,7 +1154,7 @@ def test_scalar_timestamps(c, gpu):`
`1154`	`1154`	`}`
`1155`	`1155`	`)`
`1156`	`1156`	`df1 = c.sql("SELECT FLOOR(to_timestamp(d) TO DAY) AS floor_to_day FROM df")`
`1157`		`- assert_eq(df1, expected_df)`
	`1157`	`+ assert_eq(df1, expected_df, check_dtype=(not gpu))`
`1158`	`1158`	`df2 = c.sql("SELECT FLOOR(CAST(d AS TIMESTAMP) TO DAY) AS floor_to_day FROM df")`
`1159`	`1159`	`assert_eq(df2, expected_df)`
`1160`	`1160`