Added named_arrays.AbstractArray.median() method. (#108)

sun-data · Feb 18, 2025 · 2b0d02b · 2b0d02b
1 parent 5e4e8c4
commit 2b0d02b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/named_arrays/_core.py b/named_arrays/_core.py
@@ -824,6 +824,12 @@ def var(
     ) -> Self:
         return np.var(self, axis=axis, where=where)
 
+    def median(
+            self,
+            axis: None | str | Sequence[str] = None,
+    ):
+        return np.median(self, axis=axis)
+
     def percentile(
             self: Self,
             q: int | float | u.Quantity | Self,

diff --git a/named_arrays/tests/test_core.py b/named_arrays/tests/test_core.py
@@ -1202,6 +1202,12 @@ def test_var(
     ):
         assert np.array_equal(array.var(), np.var(array))
 
+    def test_median(
+            self,
+            array: na.AbstractArray,
+    ):
+        assert np.array_equal(array.median(), np.median(array))
+
     def test_percentile(
             self,
             array: na.AbstractArray,