test setter

cmelab · Sep 19, 2023 · 27c6fb3 · 27c6fb3
1 parent 1310b79
commit 27c6fb3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cmeutils/tests/test_visualize.py b/cmeutils/tests/test_visualize.py
@@ -80,6 +80,7 @@ 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
+        assert all(np.array_equal(colors, p3ht_fresnel.colors))
 
     def test_unwrap_positions(self, p3ht_fresnel):
         assert p3ht_fresnel.unwrap_positions is False