oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#2913: Updated tests

Browse Source
This commit is contained in:
Nick Todd
2024-11-06 14:52:22 +00:00
parent 370bcfc476
commit 4c2ef6ea2a
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/integration_tests/game_layer/test_rewards.py
View File

@@ -18,7 +18,7 @@ from tests import TEST_ASSETS_ROOT
from tests.conftest import ControlledAgent from tests.conftest import ControlledAgent
def test_WebpageUnavailablePenalty(game_and_agent): def test_WebpageUnavailablePenalty(game_and_agent: tuple[PrimaiteGame, ControlledAgent]):
"""Test that we get the right reward for failing to fetch a website.""" """Test that we get the right reward for failing to fetch a website."""
# set up the scenario, configure the web browser to the correct url # set up the scenario, configure the web browser to the correct url
game, agent = game_and_agent game, agent = game_and_agent
@@ -55,7 +55,7 @@ def test_WebpageUnavailablePenalty(game_and_agent):
assert agent.reward_function.current_reward == -0.7 assert agent.reward_function.current_reward == -0.7
def test_uc2_rewards(game_and_agent): def test_uc2_rewards(game_and_agent: tuple[PrimaiteGame, ControlledAgent]):
"""Test that the reward component correctly applies a penalty when the selected client cannot reach the database.""" """Test that the reward component correctly applies a penalty when the selected client cannot reach the database."""
game, agent = game_and_agent game, agent = game_and_agent
agent: ControlledAgent agent: ControlledAgent
@@ -142,8 +142,8 @@ def test_action_penalty_loads_from_config():
act_penalty_obj = comp[0] act_penalty_obj = comp[0]
if act_penalty_obj is None: if act_penalty_obj is None:
pytest.fail("Action penalty reward component was not added to the agent from config.") pytest.fail("Action penalty reward component was not added to the agent from config.")
assert act_penalty_obj.action_penalty == -0.75 assert act_penalty_obj.config.action_penalty == -0.75
assert act_penalty_obj.do_nothing_penalty == 0.125 assert act_penalty_obj.config.do_nothing_penalty == 0.125
def test_action_penalty(): def test_action_penalty():
@@ -152,7 +152,7 @@ def test_action_penalty():
# Create an ActionPenalty Reward # Create an ActionPenalty Reward
schema = ActionPenalty.ConfigSchema(action_penalty=-0.75, do_nothing_penalty=0.125) schema = ActionPenalty.ConfigSchema(action_penalty=-0.75, do_nothing_penalty=0.125)
# Penalty = ActionPenalty(action_penalty=-0.75, do_nothing_penalty=0.125) # Penalty = ActionPenalty(action_penalty=-0.75, do_nothing_penalty=0.125)
Penalty = ActionPenalty(schema) Penalty = ActionPenalty(config=schema)
# Assert that penalty is applied if action isn't DONOTHING # Assert that penalty is applied if action isn't DONOTHING
reward_value = Penalty.calculate( reward_value = Penalty.calculate(
@@ -183,11 +183,12 @@ def test_action_penalty():
assert reward_value == 0.125 assert reward_value == 0.125
def test_action_penalty_e2e(game_and_agent): def test_action_penalty_e2e(game_and_agent: tuple[PrimaiteGame, ControlledAgent]):
"""Test that we get the right reward for doing actions to fetch a website.""" """Test that we get the right reward for doing actions to fetch a website."""
game, agent = game_and_agent game, agent = game_and_agent
agent: ControlledAgent agent: ControlledAgent
comp = ActionPenalty(action_penalty=-0.75, do_nothing_penalty=0.125) schema = ActionPenalty.ConfigSchema(action_penalty=-0.75, do_nothing_penalty=0.125)
comp = ActionPenalty(config=schema)
agent.reward_function.register_component(comp, 1.0) agent.reward_function.register_component(comp, 1.0)