Update arch for dispersion in tests

Author: ElliottKasoar

tests/test_descriptors.py

@@ -119,7 +119,7 @@ def test_dispersion():
     descriptors_disp.run()
 
     assert (
-        descriptors_disp.struct.info["mace_mp_descriptor"]
+        descriptors_disp.struct.info["mace_mp_d3_descriptor"]
         == descriptors.struct.info["mace_mp_descriptor"]
     )
 

tests/test_mlip_calculators.py

@@ -253,7 +253,7 @@ def test_d3():
     calculator = choose_calculator(arch="mace_mp", dispersion=True)
     assert calculator.parameters["version"] is not None
     assert calculator.parameters["model"] is not None
-    assert calculator.parameters["arch"] == "mace_mp-d3"
+    assert calculator.parameters["arch"] == "mace_mp_d3"
 
 
 def test_d3_manual():
@@ -264,4 +264,4 @@ def test_d3_manual():
     calculator = add_dispersion(calculator)
     assert calculator.parameters["version"] is not None
     assert calculator.parameters["model"] is not None
-    assert calculator.parameters["arch"] == "mace_mp-d3"
+    assert calculator.parameters["arch"] == "mace_mp_d3"

tests/test_singlepoint_cli.py

@@ -434,10 +434,10 @@ def test_hessian(tmp_path):
     assert result.exit_code == 0
 
     atoms = read(results_path)
-    assert "mace_mp_energy" in atoms.info
-    assert "mace_mp_hessian" in atoms.info
-    assert "mace_stress" not in atoms.info
-    assert atoms.info["mace_mp_hessian"].shape == (24, 8, 3)
+    assert "mace_mp_d3_energy" in atoms.info
+    assert "mace_mp_d3_hessian" in atoms.info
+    assert "mace_mp_d3_stress" not in atoms.info
+    assert atoms.info["mace_mp_d3_hessian"].shape == (24, 8, 3)
     assert atoms.info["units"]["hessian"] == "ev/Ang^2"