| import numpy as np | |
| from minidreamer.envs.make_env import make_env | |
| def test_make_env_returns_normalized_rgb_observation(): | |
| env = make_env(seed=0) | |
| obs, _ = env.reset() | |
| assert obs.shape == (64, 64, 3) | |
| assert obs.dtype == np.float32 | |
| assert 0.0 <= float(obs.min()) <= float(obs.max()) <= 1.0 | |
| next_obs, reward, terminated, truncated, _ = env.step(env.action_space.sample()) | |
| assert next_obs.shape == (64, 64, 3) | |
| assert isinstance(float(reward), float) | |
| assert isinstance(bool(terminated), bool) | |
| assert isinstance(bool(truncated), bool) | |
| env.close() | |