diff --git a/tests/element/testannotations.py b/tests/element/testannotations.py index a5242f94d3..530d39bb76 100644 --- a/tests/element/testannotations.py +++ b/tests/element/testannotations.py @@ -29,13 +29,13 @@ def test_text_string_position(self): def test_hline_dimension_values(self): hline = HLine(0) - self.assertEqual(hline.range(0), (np.NaN, np.NaN)) + self.assertTrue(all(not np.isfinite(v) for v in hline.range(0))) self.assertEqual(hline.range(1), (0, 0)) def test_vline_dimension_values(self): hline = VLine(0) self.assertEqual(hline.range(0), (0, 0)) - self.assertEqual(hline.range(1), (np.NaN, np.NaN)) + self.assertTrue(all(not np.isfinite(v) for v in hline.range(1))) def test_arrow_redim_range_aux(self): annotations = Arrow(0, 0)