From 6941c77b7882a12186d95fb3ffb53988085d4224 Mon Sep 17 00:00:00 2001
From: ElliottKasoar <ElliottKasoar@users.noreply.github.com>
Date: Thu, 9 May 2024 10:42:49 +0100
Subject: [PATCH] Add skip for training if MACE import fails

---
 tests/test_train_cli.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/tests/test_train_cli.py b/tests/test_train_cli.py
index 1a4ad133..752696e3 100644
--- a/tests/test_train_cli.py
+++ b/tests/test_train_cli.py
@@ -3,8 +3,16 @@
 from pathlib import Path
 import shutil
 
+import pytest
 from typer.testing import CliRunner
 
+try:
+    from mace.cli.run_train import run as run_train  # pylint: disable=unused-import
+
+    SKIP_TESTS = False
+except ImportError:
+    SKIP_TESTS = True
+
 from janus_core.cli.janus import app
 
 DATA_PATH = Path(__file__).parent / "data"
@@ -12,6 +20,7 @@
 runner = CliRunner()
 
 
+@pytest.mark.skipif(SKIP_TESTS, reason="Requires updated version of MACE")
 def test_help():
     """Test calling `janus train --help`."""
     result = runner.invoke(app, ["train", "--help"])
@@ -19,6 +28,7 @@ def test_help():
     assert "Usage: janus train [OPTIONS]" in result.stdout
 
 
+@pytest.mark.skipif(SKIP_TESTS, reason="Requires updated version of MACE")
 def test_train():
     """Test MLIP training."""
     model = "test.model"
@@ -59,6 +69,7 @@ def test_train():
         assert result.exit_code == 0
 
 
+@pytest.mark.skipif(SKIP_TESTS, reason="Requires updated version of MACE")
 def test_train_with_foundation():
     """Test MLIP training raises error with foundation_model in config."""
     config = DATA_PATH / "mlip_train_invalid.yml"
@@ -75,6 +86,7 @@ def test_train_with_foundation():
     assert isinstance(result.exception, ValueError)
 
 
+@pytest.mark.skipif(SKIP_TESTS, reason="Requires updated version of MACE")
 def test_fine_tune():
     """Test MLIP fine-tuning."""
     model = "test-finetuned.model"
@@ -111,6 +123,7 @@ def test_fine_tune():
         assert result.exit_code == 0
 
 
+@pytest.mark.skipif(SKIP_TESTS, reason="Requires updated version of MACE")
 def test_fine_tune_no_foundation():
     """Test MLIP fine-tuning raises errors without foundation_model."""
     config = DATA_PATH / "mlip_fine_tune_no_foundation.yml"
@@ -123,6 +136,7 @@ def test_fine_tune_no_foundation():
     assert isinstance(result.exception, ValueError)
 
 
+@pytest.mark.skipif(SKIP_TESTS, reason="Requires updated version of MACE")
 def test_fine_tune_invalid_foundation():
     """Test MLIP fine-tuning raises errors with invalid foundation_model."""
     config = DATA_PATH / "mlip_fine_tune_invalid_foundation.yml"