Add test for new multidiscrete spaces
This commit is contained in:
31
tests/config/box_obs_space_laydown_config.yaml
Normal file
31
tests/config/box_obs_space_laydown_config.yaml
Normal file
@@ -0,0 +1,31 @@
|
|||||||
|
- itemType: ACTIONS
|
||||||
|
type: NODE
|
||||||
|
- itemType: OBSERVATIONS
|
||||||
|
type: BOX
|
||||||
|
- itemType: STEPS
|
||||||
|
steps: 5
|
||||||
|
- itemType: PORTS
|
||||||
|
portsList:
|
||||||
|
- port: '21'
|
||||||
|
- itemType: SERVICES
|
||||||
|
serviceList:
|
||||||
|
- name: ftp
|
||||||
|
- itemType: NODE
|
||||||
|
node_id: '1'
|
||||||
|
name: node
|
||||||
|
node_class: SERVICE
|
||||||
|
node_type: COMPUTER
|
||||||
|
priority: P1
|
||||||
|
hardware_state: 'ON'
|
||||||
|
ip_address: 192.168.0.1
|
||||||
|
software_state: GOOD
|
||||||
|
file_system_state: GOOD
|
||||||
|
services:
|
||||||
|
- name: ftp
|
||||||
|
port: '21'
|
||||||
|
state: GOOD
|
||||||
|
- itemType: POSITION
|
||||||
|
positions:
|
||||||
|
- node: '1'
|
||||||
|
x_pos: 309
|
||||||
|
y_pos: 78
|
||||||
31
tests/config/multidiscrete_obs_space_laydown_config.yaml
Normal file
31
tests/config/multidiscrete_obs_space_laydown_config.yaml
Normal file
@@ -0,0 +1,31 @@
|
|||||||
|
- itemType: ACTIONS
|
||||||
|
type: NODE
|
||||||
|
- itemType: OBSERVATIONS
|
||||||
|
type: MULTIDISCRETE
|
||||||
|
- itemType: STEPS
|
||||||
|
steps: 5
|
||||||
|
- itemType: PORTS
|
||||||
|
portsList:
|
||||||
|
- port: '21'
|
||||||
|
- itemType: SERVICES
|
||||||
|
serviceList:
|
||||||
|
- name: ftp
|
||||||
|
- itemType: NODE
|
||||||
|
node_id: '1'
|
||||||
|
name: node
|
||||||
|
node_class: SERVICE
|
||||||
|
node_type: COMPUTER
|
||||||
|
priority: P1
|
||||||
|
hardware_state: 'ON'
|
||||||
|
ip_address: 192.168.0.1
|
||||||
|
software_state: GOOD
|
||||||
|
file_system_state: GOOD
|
||||||
|
services:
|
||||||
|
- name: ftp
|
||||||
|
port: '21'
|
||||||
|
state: GOOD
|
||||||
|
- itemType: POSITION
|
||||||
|
positions:
|
||||||
|
- node: '1'
|
||||||
|
x_pos: 309
|
||||||
|
y_pos: 78
|
||||||
23
tests/test_observation_space.py
Normal file
23
tests/test_observation_space.py
Normal file
@@ -0,0 +1,23 @@
|
|||||||
|
"""Test env creation and behaviour with different observation spaces."""
|
||||||
|
|
||||||
|
from tests import TEST_CONFIG_ROOT
|
||||||
|
from tests.conftest import _get_primaite_env_from_config
|
||||||
|
|
||||||
|
|
||||||
|
def test_creating_env_with_box_obs():
|
||||||
|
"""Try creating env with box observation space."""
|
||||||
|
env = _get_primaite_env_from_config(
|
||||||
|
main_config_path=TEST_CONFIG_ROOT / "one_node_states_on_off_main_config.yaml",
|
||||||
|
lay_down_config_path=TEST_CONFIG_ROOT / "box_obs_space_laydown_config.yaml",
|
||||||
|
)
|
||||||
|
env.update_environent_obs()
|
||||||
|
|
||||||
|
|
||||||
|
def test_creating_env_with_multidiscrete_obs():
|
||||||
|
"""Try creating env with MultiDiscrete observation space."""
|
||||||
|
env = _get_primaite_env_from_config(
|
||||||
|
main_config_path=TEST_CONFIG_ROOT / "one_node_states_on_off_main_config.yaml",
|
||||||
|
lay_down_config_path=TEST_CONFIG_ROOT
|
||||||
|
/ "multidiscrete_obs_space_laydown_config.yaml",
|
||||||
|
)
|
||||||
|
env.update_environent_obs()
|
||||||
Reference in New Issue
Block a user