Added named_arrays.AbstractArray.var() method. (#107)

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

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/named_arrays/_core.py b/named_arrays/_core.py
@@ -817,6 +817,13 @@ def std(
     ) -> Self:
         return np.std(self, axis=axis, where=where)
 
+    def var(
+            self: Self,
+            axis: None | str | Sequence[str] = None,
+            where: Self = np._NoValue,
+    ) -> Self:
+        return np.var(self, axis=axis, where=where)
+
     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
@@ -1196,6 +1196,12 @@ def test_std(
     ):
         assert np.array_equal(array.std(), np.std(array))
 
+    def test_var(
+            self,
+            array: na.AbstractArray,
+    ):
+        assert np.array_equal(array.var(), np.var(array))
+
     def test_percentile(
             self,
             array: na.AbstractArray,