jsonarray

Author: TomAugspurger

pandas/tests/extension/json/array.py

@@ -113,6 +113,11 @@ def __setitem__(self, key, value):
     def __len__(self) -> int:
         return len(self.data)
 
+    def __array__(self, dtype=None):
+        if dtype is None:
+            dtype = object
+        return np.asarray(self.data, dtype=dtype)
+
     @property
     def nbytes(self) -> int:
         return sys.getsizeof(self.data)

pandas/tests/extension/json/test_json.py

@@ -163,10 +163,6 @@ def test_unstack(self, data, index):
         # this matches otherwise
         return super().test_unstack(data, index)
 
-    @pytest.mark.xfail(reason="Inconsistent sizes.")
-    def test_transpose(self, data):
-        super().test_transpose(data)
-
 
 class TestGetitem(BaseJSON, base.BaseGetitemTests):
     pass