From 81710c251d77393343a461a2dac86f5a280f5c08 Mon Sep 17 00:00:00 2001 From: Sai Asish Y Date: Tue, 12 May 2026 15:55:59 -0700 Subject: [PATCH] fix: preserve units for np.positive (unary plus) --- quantities/dimensionality.py | 1 + quantities/tests/test_arithmetic.py | 14 ++++++++++++++ 2 files changed, 15 insertions(+) diff --git a/quantities/dimensionality.py b/quantities/dimensionality.py index 37fd860..fd4bb33 100644 --- a/quantities/dimensionality.py +++ b/quantities/dimensionality.py @@ -326,6 +326,7 @@ def _d_copy(q1, out=None): p_dict[np.absolute] = _d_copy p_dict[np.conjugate] = _d_copy p_dict[np.negative] = _d_copy +p_dict[np.positive] = _d_copy p_dict[np.ones_like] = _d_copy p_dict[np.rint] = _d_copy p_dict[np.floor] = _d_copy diff --git a/quantities/tests/test_arithmetic.py b/quantities/tests/test_arithmetic.py index 360db7a..cb26279 100644 --- a/quantities/tests/test_arithmetic.py +++ b/quantities/tests/test_arithmetic.py @@ -185,6 +185,20 @@ def test_negative(self): Quantity(5.0, 'm') ) + def test_positive(self): + self.assertQuantityEqual( + +pq.m, + Quantity(1, 'm') + ) + self.assertQuantityEqual( + +Quantity(5, 'm'), + Quantity(5, 'm') + ) + self.assertQuantityEqual( + +Quantity(-5.0, 'm'), + Quantity(-5.0, 'm') + ) + def test_addition(self): self.assertQuantityEqual( pq.eV + pq.eV,