Fix subset_state test for remove_subset_group

glue-viz · Jan 8, 2025 · 97ae3fd · 97ae3fd
1 parent a3aaca8
commit 97ae3fd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/glue/core/tests/test_edit_subset_mode.py b/glue/core/tests/test_edit_subset_mode.py
@@ -195,7 +195,8 @@ class Listener(HubListener):
 
     def handler(msg):
         assert len(data1.subsets) == len(data2.subsets)
-        assert data2.subsets[0].subset_state is data1.subsets[0].subset_state
+        for i in range(len(data1.subsets)):
+            assert data2.subsets[i].subset_state is data1.subsets[i].subset_state
         count[0] += 1
 
     app.session.hub.subscribe(listener, SubsetCreateMessage, handler=handler)