fix tests

metatensor · Jul 26, 2024 · 34ceaf1 · 34ceaf1
1 parent 992108a
commit 34ceaf1
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/metatrain/experimental/soap_bpnn/tests/test_functionality.py b/src/metatrain/experimental/soap_bpnn/tests/test_functionality.py
@@ -1,5 +1,7 @@
 import metatensor.torch
+import pytest
 import torch
+from jsonschema.exceptions import ValidationError
 from metatensor.torch.atomistic import ModelOutput, System
 from omegaconf import OmegaConf
 
@@ -223,6 +225,7 @@ def test_fixed_composition_weights_error():
     hypers = DEFAULT_HYPERS.copy()
     hypers["training"]["fixed_composition_weights"] = {"energy": {"H": 300.0}}
     hypers = OmegaConf.create(hypers)
-    check_architecture_options(
-        name="experimental.soap_bpnn", options=OmegaConf.to_container(hypers)
-    )
+    with pytest.raises(ValidationError, match=r"'H' does not match '\^\[0-9\]\+\$'"):
+        check_architecture_options(
+            name="experimental.soap_bpnn", options=OmegaConf.to_container(hypers)
+        )