From 2e4a1c37d1708ba7d01e3c16005f81938e1f9796 Mon Sep 17 00:00:00 2001 From: Nick Todd Date: Mon, 5 Aug 2024 10:34:06 +0100 Subject: [PATCH] #2777: Pre-commit fixes to test --- .../game_layer/test_RNG_seed.py | 33 +++++++++++-------- 1 file changed, 20 insertions(+), 13 deletions(-) diff --git a/tests/integration_tests/game_layer/test_RNG_seed.py b/tests/integration_tests/game_layer/test_RNG_seed.py index c1bb7bb0..0c6d567d 100644 --- a/tests/integration_tests/game_layer/test_RNG_seed.py +++ b/tests/integration_tests/game_layer/test_RNG_seed.py @@ -1,43 +1,50 @@ -from primaite.config.load import data_manipulation_config_path -from primaite.session.environment import PrimaiteGymEnv -from primaite.game.agent.interface import AgentHistoryItem -import yaml +# © Crown-owned copyright 2024, Defence Science and Technology Laboratory UK from pprint import pprint + import pytest +import yaml + +from primaite.config.load import data_manipulation_config_path +from primaite.game.agent.interface import AgentHistoryItem +from primaite.session.environment import PrimaiteGymEnv + @pytest.fixture() def create_env(): - with open(data_manipulation_config_path(), 'r') as f: + with open(data_manipulation_config_path(), "r") as f: cfg = yaml.safe_load(f) - env = PrimaiteGymEnv(env_config = cfg) + env = PrimaiteGymEnv(env_config=cfg) return env + def test_rng_seed_set(create_env): + """Test with RNG seed set.""" env = create_env env.reset(seed=3) for i in range(100): env.step(0) - a = [item.timestep for item in env.game.agents['client_2_green_user'].history if item.action!="DONOTHING"] + a = [item.timestep for item in env.game.agents["client_2_green_user"].history if item.action != "DONOTHING"] env.reset(seed=3) for i in range(100): env.step(0) - b = [item.timestep for item in env.game.agents['client_2_green_user'].history if item.action!="DONOTHING"] + b = [item.timestep for item in env.game.agents["client_2_green_user"].history if item.action != "DONOTHING"] + + assert a == b - assert a==b def test_rng_seed_unset(create_env): + """Test with no RNG seed.""" env = create_env env.reset() for i in range(100): env.step(0) - a = [item.timestep for item in env.game.agents['client_2_green_user'].history if item.action!="DONOTHING"] + a = [item.timestep for item in env.game.agents["client_2_green_user"].history if item.action != "DONOTHING"] env.reset() for i in range(100): env.step(0) - b = [item.timestep for item in env.game.agents['client_2_green_user'].history if item.action!="DONOTHING"] - - assert a!=b + b = [item.timestep for item in env.game.agents["client_2_green_user"].history if item.action != "DONOTHING"] + assert a != b