add tests to verify that eq always returns a ndarray

1 files changed, 12 insertions, 0 deletions

diff --git a/pint/testsuite/test_numpy.py b/pint/testsuite/test_numpy.py
index 61a8d81..0e7248b 100644
--- a/pint/testsuite/test_numpy.py
+++ b/pint/testsuite/test_numpy.py
@@ -878,11 +878,23 @@ class TestNumpyUnclassified(TestNumpyMethods):
     def test_equal(self):
         x = self.q.magnitude
         u = self.Q_(np.ones(x.shape))
+        false = np.zeros_like(x, dtype=np.bool_)
 
         self.assertQuantityEqual(u, u)
         self.assertQuantityEqual(u == u, u.magnitude == u.magnitude)
         self.assertQuantityEqual(u == 1, u.magnitude == 1)
 
+        v = self.Q_(np.zeros(x.shape), "m")
+        w = self.Q_(np.ones(x.shape), "m")
+        self.assertNDArrayEqual(v == 1, false)
+        self.assertNDArrayEqual(
+            self.Q_(np.zeros_like(x), "m") == self.Q_(np.zeros_like(x), "s"),
+            false,
+        )
+        self.assertNDArrayEqual(v == w, false)
+        self.assertNDArrayEqual(v == w.to("mm"), false)
+        self.assertNDArrayEqual(u == v, false)
+
     def test_shape(self):
         u = self.Q_(np.arange(12))
         u.shape = 4, 3