#2476 Add test for episode scheduler

tests/e2e_integration_tests/environments/test_sb3_environment.py

@@ -16,7 +16,7 @@ def test_sb3_compatibility():
     with open(data_manipulation_config_path(), "r") as f:
         cfg = yaml.safe_load(f)
 
-    gym = PrimaiteGymEnv(game_config=cfg)
+    gym = PrimaiteGymEnv(env_config=cfg)
     model = PPO("MlpPolicy", gym)
 
     model.learn(total_timesteps=1000)

tests/e2e_integration_tests/test_environment.py

@@ -21,7 +21,7 @@ class TestPrimaiteEnvironment:
         """Check that environment loads correctly from config and it can be reset."""
         with open(CFG_PATH, "r") as f:
             cfg = yaml.safe_load(f)
-        env = PrimaiteGymEnv(game_config=cfg)
+        env = PrimaiteGymEnv(env_config=cfg)
 
         def env_checks():
             assert env is not None
@@ -44,7 +44,7 @@ class TestPrimaiteEnvironment:
         """Make sure you can go all the way through the session without errors."""
         with open(CFG_PATH, "r") as f:
             cfg = yaml.safe_load(f)
-        env = PrimaiteGymEnv(game_config=cfg)
+        env = PrimaiteGymEnv(env_config=cfg)
 
         assert (num_actions := len(env.agent.action_manager.action_map)) == 54
         # run every action and make sure there's no crash
@@ -88,4 +88,4 @@ class TestPrimaiteEnvironment:
         with open(MISCONFIGURED_PATH, "r") as f:
             cfg = yaml.safe_load(f)
         with pytest.raises(pydantic.ValidationError):
-            env = PrimaiteGymEnv(game_config=cfg)
+            env = PrimaiteGymEnv(env_config=cfg)

tests/e2e_integration_tests/test_uc2_data_manipulation_scenario.py

@@ -44,7 +44,7 @@ def test_application_install_uninstall_on_uc2():
     with open(TEST_ASSETS_ROOT / "configs/test_application_install.yaml", "r") as f:
         cfg = yaml.safe_load(f)
 
-    env = PrimaiteGymEnv(game_config=cfg)
+    env = PrimaiteGymEnv(env_config=cfg)
     env.agent.flatten_obs = False
     env.reset()