fix append strng test

bluesky · Mar 14, 2024 · ac8a8ab · ac8a8ab
1 parent 9dca2d7
commit ac8a8ab
Showing 1 changed file with 17 additions and 4 deletions.
diff --git a/tiled/_tests/test_writing.py b/tiled/_tests/test_writing.py
@@ -510,9 +510,22 @@ def test_write_with_specified_mimetype(tree):
             },
         ),
         (
-            {"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"]},
+            {
+                "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]"
+                ),
+            },
         ),
     ],
 )
@@ -546,4 +559,4 @@ def test_append_partition(
 
         df3 = pandas.DataFrame(expected_file)
 
-        assert_frame_equal(x.read(), df3, check_dtype=False)
+        assert_frame_equal(x.read(), df3, check_dtype=True)