From 6d9f82a7dfe667ecba9a6d19865fcb08e7a80eb6 Mon Sep 17 00:00:00 2001 From: Keewis Date: Sun, 21 Jun 2020 13:51:12 +0200 Subject: add tests to verify that eq always returns a ndarray --- pint/testsuite/test_numpy.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'pint') 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 -- cgit v1.2.1