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

Merge remote-tracking branch 'origin/dev' into feature/2736-instantaneous-rewards

Browse Source
This commit is contained in:
Marek Wolan
2024-08-20 10:40:48 +01:00
parent 15b7334f05 b7ef538942
commit 98f5454f04
18 changed files with 1474 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -72,25 +72,26 @@ def test_uc2_rewards(game_and_agent):
request = ["network", "node", "client_1", "application", "DatabaseClient", "execute"]
response = game.simulation.apply_request(request)
state = game.get_sim_state()
reward_value = comp.calculate(
state,
last_action_response=AgentHistoryItem(
timestep=0, action="NODE_APPLICATION_EXECUTE", parameters={}, request=request, response=response
),
ahi = AgentHistoryItem(
timestep=0, action="NODE_APPLICATION_EXECUTE", parameters={}, request=request, response=response
)
reward_value = comp.calculate(state, last_action_response=ahi)
assert reward_value == 1.0
assert ahi.reward_info == {"connection_attempt_status": "success"}
router.acl.remove_rule(position=2)
response = game.simulation.apply_request(request)
state = game.get_sim_state()
ahi = AgentHistoryItem(
timestep=0, action="NODE_APPLICATION_EXECUTE", parameters={}, request=request, response=response
)
reward_value = comp.calculate(
state,
last_action_response=AgentHistoryItem(
timestep=0, action="NODE_APPLICATION_EXECUTE", parameters={}, request=request, response=response
),
last_action_response=ahi,
)
assert reward_value == -1.0
assert ahi.reward_info == {"connection_attempt_status": "failure"}
def test_shared_reward():