remove qc to_list stub and revert test_to_numpy

snowflakedb · Sep 5, 2024 · 3c2392d · 3c2392d
1 parent f72a6a4
commit 3c2392d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/integ/modin/test_to_numpy.py b/tests/integ/modin/test_to_numpy.py
@@ -58,9 +58,8 @@ def test_to_numpy_basic(data, pandas_obj, func):
             assert_array_equal(df.values, native_df.values)
     if pandas_obj == "Series":
         with SqlCounter(query_count=1):
-            res = df.to_numpy()
-        expected_res = native_df.to_numpy()
-        assert len(res) == len(expected_res)
+            res = df.to_list()
+        expected_res = native_df.to_list()
         for r1, r2 in zip(res, expected_res):
             # native pandas series returns a list of pandas Timestamp,
             # but Snowpark pandas returns a list of integers in ms.