summaryrefslogtreecommitdiff
path: root/pint/testsuite/test_numpy.py
diff options
context:
space:
mode:
Diffstat (limited to 'pint/testsuite/test_numpy.py')
-rw-r--r--pint/testsuite/test_numpy.py16
1 files changed, 16 insertions, 0 deletions
diff --git a/pint/testsuite/test_numpy.py b/pint/testsuite/test_numpy.py
index 13a773f..0efb476 100644
--- a/pint/testsuite/test_numpy.py
+++ b/pint/testsuite/test_numpy.py
@@ -1153,6 +1153,22 @@ class TestNumpyUnclassified(TestNumpyMethods):
),
) # Note: Does not support Quantity pad_with vectorized callable use
+ @helpers.requires_array_function_protocol()
+ def test_allclose(self):
+ self.assertTrue(
+ np.allclose([1e10, 1e-8] * self.ureg.m, [1.00001e10, 1e-9] * self.ureg.m)
+ )
+ self.assertFalse(
+ np.allclose([1e10, 1e-8] * self.ureg.m, [1.00001e10, 1e-9] * self.ureg.mm)
+ )
+
+ @helpers.requires_array_function_protocol()
+ def test_intersect1d(self):
+ self.assertQuantityEqual(
+ np.intersect1d([1, 3, 4, 3] * self.ureg.m, [3, 1, 2, 1] * self.ureg.m),
+ [1, 3] * self.ureg.m,
+ )
+
@unittest.skip
class TestBitTwiddlingUfuncs(TestUFuncs):