minor

Author: chelsea-lin

db_dtypes/json.py

@@ -202,18 +202,6 @@ def dtype(self) -> JSONDtype:
     def __contains__(self, key) -> bool:
         return super().__contains__(JSONArray._seralizate_json(key))
 
-    # def __contains__(self, key) -> bool:
-    #     # https://github.com/pandas-dev/pandas/pull/51307#issuecomment-1426372604
-    #     if pd.isna(key) and key is not self.dtype.na_value:
-    #         if self.dtype.kind == "f" and lib.is_float(key):
-    #             return pc.any(pc.is_nan(self._pa_array)).as_py()
-
-    #         # e.g. date or timestamp types we do not allow None here to match pd.NA
-    #         return False
-    #         # TODO: maybe complex? object?
-
-    #     return bool(super().__contains__(key))
-
     def insert(self, loc: int, item) -> JSONArray:
         val = JSONArray._seralizate_json(item)
         return super().insert(loc, val)

tests/compliance/json/test_json_compliance.py

@@ -402,45 +402,3 @@ def test_getitem_scalar(self, data):
 
         result = pd.Series(data)[0]
         assert isinstance(result, dict)
-
-
-def custom_assert_frame_equal(left, right, *args, **kwargs):
-    obj_type = kwargs.get("obj", "DataFrame")
-    tm.assert_index_equal(
-        left.columns,
-        right.columns,
-        exact=kwargs.get("check_column_type", "equiv"),
-        check_names=kwargs.get("check_names", True),
-        check_exact=kwargs.get("check_exact", False),
-        check_categorical=kwargs.get("check_categorical", True),
-        obj=f"{obj_type}.columns",
-    )
-
-    jsons = (left.dtypes == "json").index
-
-    for col in jsons:
-        tm.assert_series_equal(left[col], right[col], *args, **kwargs)
-
-    left = left.drop(columns=jsons)
-    right = right.drop(columns=jsons)
-    tm.assert_frame_equal(left, right, *args, **kwargs)
-
-
-def test_custom_asserts():
-    data = JSONArray._from_sequence(
-        [
-            {"a": 1},
-            {"b": 2},
-            {"c": 3},
-        ]
-    )
-    a = pd.Series(data)
-    tm.assert_series_equal(a, a)
-    custom_assert_frame_equal(a.to_frame(), a.to_frame())
-
-    b = pd.Series(data.take([0, 0, 1]))
-    with pytest.raises(AssertionError):
-        tm.assert_series_equal(a, b)
-
-    with pytest.raises(AssertionError):
-        custom_assert_frame_equal(a.to_frame(), b.to_frame())