Cleanup tests

danepitkin · Aug 23, 2024 · 0f6072d · 0f6072d
1 parent 06dfe49
commit 0f6072d
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/python/pyarrow/tests/test_table.py b/python/pyarrow/tests/test_table.py
@@ -3406,12 +3406,15 @@ def cpu_and_cuda_chunked_array(cpu_arrays, cuda_arrays):
 def test_chunked_array_non_cpu(cuda_context, cpu_chunked_array, cuda_chunked_array,
                                cpu_and_cuda_chunked_array):
     # type test
-    assert cuda_chunked_array.type == pa.int32()
+    assert cuda_chunked_array.type == cpu_chunked_array.type
 
     # length() test
-    assert cuda_chunked_array.length() == 10
+    assert cuda_chunked_array.length() == cpu_chunked_array.length()
 
-    # str(), repr() test
+    # str() test
+    assert str(cuda_chunked_array) == str(cpu_chunked_array)
+
+    # repr() test
     assert str(cuda_chunked_array) in repr(cuda_chunked_array)
 
     # validate() test