changed the type check in assert_frame_equal

bluesky · Mar 15, 2024 · 62eeab5 · 62eeab5
1 parent ac8a8ab
commit 62eeab5
Showing 1 changed file with 4 additions and 17 deletions.
diff --git a/tiled/_tests/test_writing.py b/tiled/_tests/test_writing.py
@@ -510,22 +510,9 @@ def test_write_with_specified_mimetype(tree):
             },
         ),
         (
-            {
-                "C": pandas.array(["x", "y"], dtype="string[pyarrow]"),
-                "D": pandas.array(["a", "b"], dtype="string[pyarrow]"),
-            },
-            {
-                "C": pandas.array(["xx", "yy", "zz"], dtype="string[pyarrow]"),
-                "D": pandas.array(["aa", "bb", "cc"], dtype="string[pyarrow]"),
-            },
-            {
-                "C": pandas.array(
-                    ["x", "y", "xx", "yy", "zz"], dtype="string[pyarrow]"
-                ),
-                "D": pandas.array(
-                    ["a", "b", "aa", "bb", "cc"], dtype="string[pyarrow]"
-                ),
-            },
+            {"C": ["x", "y"], "D": ["a", "b"]},
+            {"C": ["xx", "yy", "zz"], "D": ["aa", "bb", "cc"]},
+            {"C": ["x", "y", "xx", "yy", "zz"], "D": ["a", "b", "aa", "bb", "cc"]},
         ),
     ],
 )
@@ -559,4 +546,4 @@ def test_append_partition(
 
         df3 = pandas.DataFrame(expected_file)
 
-        assert_frame_equal(x.read(), df3, check_dtype=True)
+        assert_frame_equal(x.read(), df3, check_dtype=False)