Give the UC2 config load function a meaningful name

This commit is contained in:
Marek Wolan
2024-03-04 19:43:51 +00:00
parent 1e8dfa40cf
commit a222a8c58f
11 changed files with 21 additions and 21 deletions

View File

@@ -4,7 +4,7 @@ import yaml
from ray import air, tune
from ray.rllib.algorithms.ppo import PPOConfig
from primaite.config.load import example_config_path
from primaite.config.load import data_manipulation_config_path
from primaite.game.game import PrimaiteGame
from primaite.session.environment import PrimaiteRayMARLEnv
@@ -13,7 +13,7 @@ from primaite.session.environment import PrimaiteRayMARLEnv
def test_rllib_multi_agent_compatibility():
"""Test that the PrimaiteRayEnv class can be used with a multi agent RLLIB system."""
with open(example_config_path(), "r") as f:
with open(data_manipulation_config_path(), "r") as f:
cfg = yaml.safe_load(f)
game = PrimaiteGame.from_config(cfg)

View File

@@ -6,7 +6,7 @@ import ray
import yaml
from ray.rllib.algorithms import ppo
from primaite.config.load import example_config_path
from primaite.config.load import data_manipulation_config_path
from primaite.game.game import PrimaiteGame
from primaite.session.environment import PrimaiteRayEnv
@@ -14,7 +14,7 @@ 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:
with open(data_manipulation_config_path(), "r") as f:
cfg = yaml.safe_load(f)
game = PrimaiteGame.from_config(cfg)

View File

@@ -6,14 +6,14 @@ import pytest
import yaml
from stable_baselines3 import PPO
from primaite.config.load import example_config_path
from primaite.config.load import data_manipulation_config_path
from primaite.game.game import PrimaiteGame
from primaite.session.environment import PrimaiteGymEnv
def test_sb3_compatibility():
"""Test that the Gymnasium environment can be used with an SB3 agent."""
with open(example_config_path(), "r") as f:
with open(data_manipulation_config_path(), "r") as f:
cfg = yaml.safe_load(f)
gym = PrimaiteGymEnv(game_config=cfg)

View File

@@ -1,4 +1,4 @@
from primaite.config.load import example_config_path
from primaite.config.load import data_manipulation_config_path
from primaite.simulator.network.container import Network
from primaite.simulator.network.hardware.node_operating_state import NodeOperatingState
from primaite.simulator.network.hardware.nodes.host.computer import Computer
@@ -7,7 +7,7 @@ from tests.integration_tests.configuration_file_parsing import BASIC_CONFIG, DMZ
def test_example_config():
"""Test that the example config can be parsed properly."""
game = load_config(example_config_path())
game = load_config(data_manipulation_config_path())
network: Network = game.simulation.network
assert len(network.nodes) == 10 # 10 nodes in example network

View File

@@ -4,7 +4,7 @@ from typing import Union
import yaml
from primaite.config.load import example_config_path
from primaite.config.load import data_manipulation_config_path
from primaite.game.agent.data_manipulation_bot import DataManipulationAgent
from primaite.game.agent.interface import ProxyAgent, RandomAgent
from primaite.game.game import APPLICATION_TYPES_MAPPING, PrimaiteGame, SERVICE_TYPES_MAPPING
@@ -37,7 +37,7 @@ def load_config(config_path: Union[str, Path]) -> PrimaiteGame:
def test_example_config():
"""Test that the example config can be parsed properly."""
game = load_config(example_config_path())
game = load_config(data_manipulation_config_path())
assert len(game.agents) == 4 # red, blue and 2 green agents