fix: address failing 2D array compliance tests in DateArray

db_dtypes/core.py

@@ -135,29 +135,35 @@ def min(self, *, axis: Optional[int] = None, skipna: bool = True, **kwargs):
         result = pandas_backports.nanmin(
             values=self._ndarray, axis=axis, mask=self.isna(), skipna=skipna
         )
-        return self._box_func(result)
+        if axis is None or self.ndim == 1:
+            return self._box_func(result)
+        return self._from_backing_data(result)
 
     def max(self, *, axis: Optional[int] = None, skipna: bool = True, **kwargs):
         pandas_backports.numpy_validate_max((), kwargs)
         result = pandas_backports.nanmax(
             values=self._ndarray, axis=axis, mask=self.isna(), skipna=skipna
         )
-        return self._box_func(result)
-
-    if pandas_release >= (1, 2):
-
-        def median(
-            self,
-            *,
-            axis: Optional[int] = None,
-            out=None,
-            overwrite_input: bool = False,
-            keepdims: bool = False,
-            skipna: bool = True,
-        ):
-            pandas_backports.numpy_validate_median(
-                (),
-                {"out": out, "overwrite_input": overwrite_input, "keepdims": keepdims},
-            )
-            result = pandas_backports.nanmedian(self._ndarray, axis=axis, skipna=skipna)
+        if axis is None or self.ndim == 1:
+            return self._box_func(result)
+        return self._from_backing_data(result)
+
+    def median(
+        self,
+        *,
+        axis: Optional[int] = None,
+        out=None,
+        overwrite_input: bool = False,
+        keepdims: bool = False,
+        skipna: bool = True,
+    ):
+        if not hasattr(pandas_backports, "numpy_validate_median"):
+            raise NotImplementedError("Need pandas 1.3 or later to calculate median.")
+
+        pandas_backports.numpy_validate_median(
+            (), {"out": out, "overwrite_input": overwrite_input, "keepdims": keepdims},
+        )
+        result = pandas_backports.nanmedian(self._ndarray, axis=axis, skipna=skipna)
+        if axis is None or self.ndim == 1:
             return self._box_func(result)
+        return self._from_backing_data(result)

tests/unit/test_date.py

@@ -65,3 +65,17 @@ def test_date_parsing(value, expected):
 def test_date_parsing_errors(value, error):
     with pytest.raises(ValueError, match=error):
         pandas.Series([value], dtype="dbdate")
+
+
+# TODO: skip if median not available
+@pytest.mark.parametrize(
+    "values, expected",
+    [
+        (["1970-01-01", "1900-01-01", "2000-01-01"], datetime.date(1970, 1, 1)),
+        ([None, "1900-01-01", None], datetime.date(1900, 1, 1)),
+        (["2222-02-01", "2222-02-03"], datetime.date(2222, 2, 2)),
+    ],
+)
+def test_date_median(values, expected):
+    series = pandas.Series(values, dtype="dbdate")
+    assert series.median() == expected

tests/unit/test_date_compliance.py

@@ -0,0 +1,51 @@
+# Copyright 2022 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""
+Tests for extension interface compliance, inherited from pandas.
+
+See:
+https://github.com/pandas-dev/pandas/blob/main/pandas/tests/extension/test_period.py
+"""
+
+import datetime
+
+import numpy
+from pandas.tests.extension import base
+import pytest
+
+from db_dtypes import DateArray
+
+# NDArrayBacked2DTests suite added in https://github.com/pandas-dev/pandas/pull/44974
+pytest.importorskip("pandas", minversion="1.5.0dev")
+
+
+@pytest.fixture
+def data():
+    return DateArray(
+        numpy.arange(
+            datetime.datetime(1900, 1, 1),
+            datetime.datetime(2099, 12, 31),
+            datetime.timedelta(days=13),
+            dtype="datetime64[ns]",
+        )
+    )
+
+
+@pytest.fixture
+def data_missing():
+    return DateArray([None, datetime.date(2022, 1, 27)])
+
+
+class Test2DCompat(base.NDArrayBacked2DTests):
+    pass