offsets as deltas

hgrecco · May 21, 2024 · 655e3da · 655e3da
1 parent b161850
commit 655e3da
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/pint/facets/numpy/numpy_func.py b/pint/facets/numpy/numpy_func.py
@@ -1004,16 +1004,16 @@ def implementation(a, *args, **kwargs):
 
 # Handle functions with output unit defined by operation
 for func_str in (
-    "std",
-    "nanstd",
     "sum",
     "nansum",
     "cumsum",
     "nancumsum",
     "linalg.norm",
 ):
     implement_func("function", func_str, input_units=None, output_unit="sum")
-for func_str in ("diff", "ediff1d"):
+for func_str in ("diff", "ediff1d",
+    "std",
+    "nanstd",):
     implement_func("function", func_str, input_units=None, output_unit="delta")
 for func_str in ("gradient",):
     implement_func("function", func_str, input_units=None, output_unit="delta,div")

diff --git a/pint/testsuite/test_numpy.py b/pint/testsuite/test_numpy.py
@@ -850,11 +850,14 @@ def test_nanmedian_numpy_func(self):
         assert np.nanmedian(self.q_nan) == 2 * self.ureg.m
 
     def test_var(self):
+        assert self.q.var() == 1.25 * self.ureg.m**2
         assert self.q_temperature.var() == 1.25 * self.ureg.delta_degC**2
 
     @helpers.requires_array_function_protocol()
     def test_var_numpy_func(self):
         assert np.var(self.q) == 1.25 * self.ureg.m**2
+        assert np.var(self.q_temperature) == 1.25 * self.ureg.delta_degC**2
+
 
     @helpers.requires_array_function_protocol()
     def test_nanvar_numpy_func(self):