#2350: add tests to check spaces + acl obs test + nmne space changes

This commit is contained in:
Czar Echavez
2024-03-11 17:47:33 +00:00
parent a228a09917
commit cd6d6325db
9 changed files with 193 additions and 12 deletions

View File

@@ -2,6 +2,7 @@ import copy
from uuid import uuid4
import pytest
from gymnasium import spaces
from primaite.game.agent.observations.node_observations import NodeObservation
from primaite.simulator.network.hardware.nodes.host.computer import Computer
@@ -24,6 +25,8 @@ def test_node_observation(simulation):
node_obs = NodeObservation(where=["network", "nodes", pc.hostname])
assert node_obs.space["operating_status"] == spaces.Discrete(5)
observation_state = node_obs.observe(simulation.describe_state())
assert observation_state.get("operating_status") == 1 # computer is on