From a283f0fdaad778ff153f2fc9c23d72341460465c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jules=20Ch=C3=A9ron?= Date: Sat, 29 Apr 2023 00:01:17 +0200 Subject: fix(test): Add tests for all close with nan and atol default --- pint/testsuite/test_numpy.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/pint/testsuite/test_numpy.py b/pint/testsuite/test_numpy.py index 080c60a..f9f1095 100644 --- a/pint/testsuite/test_numpy.py +++ b/pint/testsuite/test_numpy.py @@ -1369,6 +1369,9 @@ class TestNumpyUnclassified(TestNumpyMethods): @helpers.requires_array_function_protocol() def test_allclose(self): assert np.allclose([1e10, 1e-8] * self.ureg.m, [1.00001e10, 1e-9] * self.ureg.m) + assert np.allclose( + [1e10, 1e-8] * self.ureg.m, [1.00001e13, 1e-6] * self.ureg.mm + ) assert not np.allclose( [1e10, 1e-8] * self.ureg.m, [1.00001e10, 1e-9] * self.ureg.mm ) @@ -1378,6 +1381,12 @@ class TestNumpyUnclassified(TestNumpyMethods): atol=1e-8 * self.ureg.m, ) + assert not np.allclose([1.0, np.nan] * self.ureg.m, [1.0, np.nan] * self.ureg.m) + + assert np.allclose( + [1.0, np.nan] * self.ureg.m, [1.0, np.nan] * self.ureg.m, equal_nan=True + ) + with pytest.raises(DimensionalityError): assert np.allclose( [1e10, 1e-8] * self.ureg.m, [1.00001e10, 1e-9] * self.ureg.m, atol=1e-8 -- cgit v1.2.1