update tests to reflect changes in behaviour

deruyter92 · deruyter92 · commit 9ce60b50aacc · 2026-02-03T09:31:19.000+01:00
diff --git a/tests/core/config/test_core_config.py b/tests/core/config/test_core_config.py
@@ -10,6 +10,7 @@
 #
 """Tests for deeplabcut.core.config."""
 from pathlib import Path
+from typing import Mapping
 
 import pytest
 
@@ -115,7 +116,7 @@ def capture(s):
 @pytest.mark.parametrize("multianimal", [False, True])
 def test_create_config_template_returns_tuple(multianimal):
     cfg_file, ruamel_file = create_config_template(multianimal=multianimal)
-    assert isinstance(cfg_file, dict)
+    assert isinstance(cfg_file, Mapping)
     assert ruamel_file is not None
 
 
@@ -128,9 +129,6 @@ def test_create_config_template_single_animal_has_expected_keys():
     assert "video_sets" in cfg_file
     assert "multianimalproject" in cfg_file
     assert "detector_batch_size" in cfg_file
-    assert "individuals" not in cfg_file
-    assert "uniquebodyparts" not in cfg_file
-    assert "multianimalbodyparts" not in cfg_file
 
 
 def test_create_config_template_multianimal_has_extra_keys():
@@ -148,7 +146,7 @@ def test_create_config_template_multianimal_has_extra_keys():
 
 def test_create_config_template_3d_returns_tuple():
     cfg_file_3d, ruamel_file_3d = create_config_template_3d()
-    assert isinstance(cfg_file_3d, dict)
+    assert isinstance(cfg_file_3d, Mapping)
     assert ruamel_file_3d is not None
 
 
@@ -168,19 +166,19 @@ def test_create_config_template_3d_has_expected_keys():
 
 
 def test_read_config_raises_when_file_missing(tmp_path):
-    with pytest.raises(FileNotFoundError, match="Config file at .* not found"):
+    with pytest.raises(FileNotFoundError):
         read_config(tmp_path / "missing.yaml")
 
 
 def test_read_config_sets_missing_engine_and_writes_back(tmp_path):
     config_path = tmp_path / "config.yaml"
     config_path.write_text("project_path: /other/path\n")
     cfg = read_config(config_path)
-    assert cfg["engine"] == "tensorflow"
+    assert cfg["engine"] == "pytorch"
     assert cfg["project_path"] == str(tmp_path)
     # File should have been updated
     cfg_again = read_config_as_dict(config_path)
-    assert cfg_again["engine"] == "tensorflow"
+    assert cfg_again["engine"] == "pytorch"
 
 
 def test_read_config_sets_detector_snapshotindex_when_missing(tmp_path):
