test_observation_wrappers.py 文件源码

def test_flattened_wrapper():
    expect = gym.make("ProvideTest-v0")
    md = spaces.MultiDiscrete([(0, 1), (0, 1)])
    expect.observation_space = md
    expect.provide_observation  = (1, 1)
    wrapper = FlattenedObservationWrapper(expect)
    o, r, d, i = wrapper.step(3)
    assert wrapper.observation_space.contains(o)
    assert o == 3