add unit test for setting colors

cmelab · Sep 19, 2023 · 1310b79 · 1310b79
1 parent a4b3da3
commit 1310b79
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmeutils/tests/test_visualize.py b/cmeutils/tests/test_visualize.py
@@ -76,6 +76,11 @@ def test_set_bad_unwrap_pos(self, p3ht_fresnel):
         with pytest.raises(ValueError):
             p3ht_fresnel.unwrap_positions = "true"
 
+    def set_set_colors(self, p3ht_fresnel):
+        colors = np.zeros_like(p3ht_fresnel.positions)
+        colors[:] = np.array([0.5, 0.5, 0.5])
+        p3ht_fresnel.colors = colors
+
     def test_unwrap_positions(self, p3ht_fresnel):
         assert p3ht_fresnel.unwrap_positions is False
         p3ht_fresnel.unwrap_positions = True