try fix test_large_pd

kuzudb · Nov 20, 2023 · 55cd3d6 · 55cd3d6
1 parent 4240f2f
commit 55cd3d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/python_api/test/test_scan_pandas.py b/tools/python_api/test/test_scan_pandas.py
@@ -84,7 +84,7 @@ def test_large_pd(get_tmp_path):
     num_rows = 40000
     odd_numbers = [2 * i + 1 for i in range(num_rows)]
     even_numbers = [2 * i for i in range(num_rows)]
-    df = pd.DataFrame({'odd': odd_numbers, 'even': even_numbers})
+    df = pd.DataFrame({'odd': np.array(odd_numbers, dtype=np.int64), 'even': np.array(even_numbers, dtype=np.int64)})
     result = conn.execute("CALL READ_PANDAS('df') RETURN *").get_as_df()
     assert result['odd'].to_list() == odd_numbers
     assert result['even'].to_list() == even_numbers