Fix tests 2

Author: alejoe91

src/spikeinterface/curation/tests/test_curation_model.py

@@ -91,8 +91,8 @@ def test_merge_units():
         "format_version": "2",
         "unit_ids": [1, 2, 3, 4],
         "merges": [
-            {"merge_unit_group": [1, 2], "merge_new_unit_id": 5},
-            {"merge_unit_group": [3, 4], "merge_new_unit_id": 6},
+            {"unit_ids": [1, 2], "new_unit_id": 5},
+            {"unit_ids": [3, 4], "new_unit_id": 6},
         ],
     }
 
@@ -122,7 +122,7 @@ def test_merge_units():
     invalid_merge_group = {
         "format_version": "2",
         "unit_ids": [1, 2, 3],
-        "merges": [{"merge_unit_group": [1], "merge_new_unit_id": 4}],
+        "merges": [{"unit_ids": [1], "new_unit_id": 4}],
     }
     with pytest.raises(ValidationError):
         CurationModel(**invalid_merge_group)
@@ -132,8 +132,8 @@ def test_merge_units():
         "format_version": "2",
         "unit_ids": [1, 2, 3],
         "merges": [
-            {"merge_unit_group": [1, 2], "merge_new_unit_id": 4},
-            {"merge_unit_group": [2, 3], "merge_new_unit_id": 5},
+            {"unit_ids": [1, 2], "new_unit_id": 4},
+            {"unit_ids": [2, 3], "new_unit_id": 5},
         ],
     }
     with pytest.raises(ValidationError):
@@ -149,9 +149,9 @@ def test_split_units():
         "splits": [
             {
                 "unit_id": 1,
-                "split_mode": "indices",
-                "split_indices": [[0, 1, 2], [3, 4, 5]],
-                "split_new_unit_ids": [4, 5],
+                "mode": "indices",
+                "indices": [[0, 1, 2], [3, 4, 5]],
+                "new_unit_ids": [4, 5],
             }
         ],
     }
@@ -165,9 +165,7 @@ def test_split_units():
     valid_split_labels = {
         "format_version": "2",
         "unit_ids": [1, 2, 3],
-        "splits": [
-            {"unit_id": 1, "split_mode": "labels", "split_labels": [0, 0, 1, 1, 0, 2], "split_new_unit_ids": [4, 5, 6]}
-        ],
+        "splits": [{"unit_id": 1, "mode": "labels", "labels": [0, 0, 1, 1, 0, 2], "new_unit_ids": [4, 5, 6]}],
     }
 
     model = CurationModel(**valid_split_labels)
@@ -193,7 +191,7 @@ def test_split_units():
     invalid_unit_id = {
         "format_version": "2",
         "unit_ids": [1, 2, 3],
-        "splits": [{"unit_id": 4, "split_mode": "indices", "split_indices": [[0, 1], [2, 3]]}],  # Non-existent unit
+        "splits": [{"unit_id": 4, "mode": "indices", "indices": [[0, 1], [2, 3]]}],  # Non-existent unit
     }
     with pytest.raises(ValidationError):
         CurationModel(**invalid_unit_id)
@@ -205,9 +203,9 @@ def test_split_units():
         "splits": [
             {
                 "unit_id": 1,
-                "split_mode": "indices",
-                "split_indices": [[0, 1], [2, 3]],
-                "split_new_unit_ids": [4],  # Should have 2 new IDs for 2 splits
+                "mode": "indices",
+                "indices": [[0, 1], [2, 3]],
+                "new_unit_ids": [4],  # Should have 2 new IDs for 2 splits
             }
         ],
     }
@@ -231,7 +229,7 @@ def test_removed_units():
     invalid_merge_remove = {
         "format_version": "2",
         "unit_ids": [1, 2, 3],
-        "merges": [{"merge_unit_group": [1, 2], "merge_new_unit_id": 4}],
+        "merges": [{"unit_ids": [1, 2], "new_unit_id": 4}],
         "removed": [1],  # Unit is both merged and removed
     }
     with pytest.raises(ValidationError):
@@ -248,10 +246,8 @@ def test_complete_model():
             "tags": LabelDefinition(name="tags", label_options=["burst", "slow"], exclusive=False),
         },
         "manual_labels": [{"unit_id": 1, "labels": {"quality": ["good"], "tags": ["burst"]}}],
-        "merges": [{"merge_unit_group": [2, 3], "merge_new_unit_id": 6}],
-        "splits": [
-            {"unit_id": 4, "split_mode": "indices", "split_indices": [[0, 1], [2, 3]], "split_new_unit_ids": [7, 8]}
-        ],
+        "merges": [{"unit_ids": [2, 3], "new_unit_id": 6}],
+        "splits": [{"unit_id": 4, "mode": "indices", "indices": [[0, 1], [2, 3]], "new_unit_ids": [7, 8]}],
         "removed": [5],
     }