def test_eager_observation_space(env: LlvmEnv):
env.observation_space = "Autophase"
assert env.observation_space.id == "Autophase"
env.observation_space = None
assert env.observation_space is None
invalid = "invalid value"
with pytest.raises(LookupError) as ctx:
env.observation_space = invalid
assert str(ctx.value) == f"Observation space not found: {invalid}"
def test_default_observation_space(env: LlvmEnv):
env.observation_space = "Autophase"
assert env.observation_space.shape == (56, )
assert env.observation_space_spec.id == "Autophase"
env.observation_space = None
assert env.observation_space is None
assert env.observation_space_spec is None
invalid = "invalid value"
with pytest.raises(LookupError,
match=f"Observation space not found: {invalid}"):
env.observation_space = invalid
def test_set_observation_space_from_spec(env: LlvmEnv):
env.observation_space = env.observation.spaces["Autophase"]
obs = env.observation_space
env.observation_space = "Autophase"
assert env.observation_space == obs
def test_autophase_observation_space_reset(env: LlvmEnv):
"""Test that the intial observation is returned on env.reset()."""
env.observation_space = "Autophase"
observation = env.reset("cbench-v1/crc32")
print(observation.tolist()) # For debugging on error.
np.testing.assert_array_equal(observation, AUTOPHASE_CBENCH_CRC32)
