24 lines
896 B
Python
24 lines
896 B
Python
"""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()
|