| import numpy as np |
| from spectral_world_models.v14_external_control import rgb_to_gray32, build_vocab, obs_to_tokens, ExternalConfig, SUPPORTED_ENVS |
|
|
| def test_external_suite_declared(): |
| assert set(SUPPORTED_ENVS)=={"CartPole-v1","MountainCar-v0","Acrobot-v1"} |
|
|
| def test_render_preprocess_shape_range(): |
| rgb=np.zeros((40,60,3),dtype=np.uint8); rgb[10:20,20:30]=255 |
| x=rgb_to_gray32(rgb,32); assert x.shape==(1,32,32); assert 0<=x.min()<=x.max()<=1 |
|
|
| def test_external_tokens_bounded(): |
| obs=np.array([0.,1.,-2.],dtype=np.float32); low=np.array([-1.,-np.inf,-3.]); high=np.array([1.,np.inf,3.]) |
| v=build_vocab('X',3); t=obs_to_tokens(obs,low,high,'X',8); assert t.shape==(8,); assert int(t.max())<len(v) |
|
|
