From 64c7dd3c84394e3f5831d55d22939ddb3c9b0b71 Mon Sep 17 00:00:00 2001
From: Marek Wolan <marek.wolan@methods.co.uk>
Date: Fri, 24 Nov 2023 12:03:46 +0000
Subject: [PATCH] Skip slow tests for now.

---
 .../environments/test_rllib_multi_agent_environment.py          | 2 ++
 .../environments/test_rllib_single_agent_environment.py         | 2 ++
 tests/e2e_integration_tests/test_primaite_session.py            | 1 +
 3 files changed, 5 insertions(+)

diff --git a/tests/e2e_integration_tests/environments/test_rllib_multi_agent_environment.py b/tests/e2e_integration_tests/environments/test_rllib_multi_agent_environment.py
index 0cf245b4..3934ce5b 100644
--- a/tests/e2e_integration_tests/environments/test_rllib_multi_agent_environment.py
+++ b/tests/e2e_integration_tests/environments/test_rllib_multi_agent_environment.py
@@ -1,3 +1,4 @@
+import pytest
 import ray
 import yaml
 from ray import air, tune
@@ -8,6 +9,7 @@ from primaite.game.game import PrimaiteGame
 from primaite.session.environment import PrimaiteRayMARLEnv
 
 
+@pytest.mark.skip(reason="Slow, reenable later")
 def test_rllib_multi_agent_compatibility():
     """Test that the PrimaiteRayEnv class can be used with a multi agent RLLIB system."""
 
diff --git a/tests/e2e_integration_tests/environments/test_rllib_single_agent_environment.py b/tests/e2e_integration_tests/environments/test_rllib_single_agent_environment.py
index ce23501a..2b12ad98 100644
--- a/tests/e2e_integration_tests/environments/test_rllib_single_agent_environment.py
+++ b/tests/e2e_integration_tests/environments/test_rllib_single_agent_environment.py
@@ -1,6 +1,7 @@
 import tempfile
 from pathlib import Path
 
+import pytest
 import ray
 import yaml
 from ray.rllib.algorithms import ppo
@@ -10,6 +11,7 @@ from primaite.game.game import PrimaiteGame
 from primaite.session.environment import PrimaiteRayEnv
 
 
+@pytest.mark.skip(reason="Slow, reenable later")
 def test_rllib_single_agent_compatibility():
     """Test that the PrimaiteRayEnv class can be used with a single agent RLLIB system."""
     with open(example_config_path(), "r") as f:
diff --git a/tests/e2e_integration_tests/test_primaite_session.py b/tests/e2e_integration_tests/test_primaite_session.py
index 5ca99cfc..086e9af8 100644
--- a/tests/e2e_integration_tests/test_primaite_session.py
+++ b/tests/e2e_integration_tests/test_primaite_session.py
@@ -65,6 +65,7 @@ class TestPrimaiteSession:
             session.start_session()
             # TODO: include checks that the model was loaded and that the eval-only session ran
 
+    @pytest.mark.skip(reason="Slow, reenable later")
     @pytest.mark.parametrize("temp_primaite_session", [[MULTI_AGENT_PATH]], indirect=True)
     def test_multi_agent_session(self, temp_primaite_session):
         """Check that we can run a training session with a multi agent system."""