Add support for pytest.approx comparisons between array and scalar

2018-03-14 15:29:40 -03:00
parent cbb2c55dea
commit c34dde7a3f
3 changed files with 33 additions and 4 deletions
--- a/testing/python/approx.py
+++ b/testing/python/approx.py
@@ -391,3 +391,14 @@ class TestApprox(object):
        """
        with pytest.raises(TypeError):
            op(1, approx(1, rel=1e-6, abs=1e-12))
+
+    def test_numpy_array_with_scalar(self):
+        np = pytest.importorskip('numpy')
+
+        actual = np.array([1 + 1e-7, 1 - 1e-8])
+        expected = 1.0
+
+        assert actual == approx(expected, rel=5e-7, abs=0)
+        assert actual != approx(expected, rel=5e-8, abs=0)
+        assert approx(expected, rel=5e-7, abs=0) == actual
+        assert approx(expected, rel=5e-8, abs=0) != actual