test_color_generator() now runs on numpy versions <=1.25.0

GuyTeichman · Oct 8, 2023 · 2f46326 · 2f46326
1 parent f019b07
commit 2f46326
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_generic.py b/tests/test_generic.py
@@ -42,7 +42,8 @@ def test_color_generator():
                      'lawngreen', 'moccasin', 'thistle']
     for i in range(150):
         color = next(gen)
-        assert color in preset_colors or (isinstance(color, np.ndarray) and len(color) == 3 and
+        assert (isinstance(color, str) and color in preset_colors) or (
+                isinstance(color, np.ndarray) and len(color) == 3 and
                                           np.max(color) <= 1 and np.min(color) >= 0)