Fixed unit tests

holoviz · Jul 4, 2018 · 71cf6d9 · 71cf6d9
1 parent 6ed529a
commit 71cf6d9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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.assertFalse(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))
+        vline = VLine(0)
+        self.assertFalse(all(not np.isfinite(v) for v in vline.range(0)))
+        self.assertEqual(vline.range(1), (np.NaN, np.NaN))
 
     def test_arrow_redim_range_aux(self):
         annotations = Arrow(0, 0)