Update test_utils.py

PennLINC · Jan 17, 2025 · 6c3f3e1 · 6c3f3e1
1 parent 45bcc76
commit 6c3f3e1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/cubids/tests/test_utils.py b/cubids/tests/test_utils.py
@@ -142,8 +142,7 @@ def test_format_params():
 
 
 def compare_group_assignments(list1, list2):
-    """
-    Compare two lists for equality based on group assignments.
+    """Compare two lists for equality based on group assignments.
 
     This function checks if two lists can be considered equal based on their group assignments.
     The actual values in the lists do not matter, only the group assignments do. Each unique value
@@ -169,6 +168,11 @@ def compare_group_assignments(list1, list2):
     >>> compare_group_assignments(list1, list2)
     True
 
+    >>> list1 = [1, 2, 1, 3, 2]
+    >>> list2 = ['b', 'd', 'b', 'q', 'd']
+    >>> compare_group_assignments(list1, list2)
+    True
+
     >>> list1 = [1, 2, 1, 3, 2]
     >>> list2 = ['a', 'b', 'a', 'c', 'd']
     >>> compare_group_assignments(list1, list2)