def test_copy_task_action_observation_space(self):
    """Test that sequence data env returns correct action/observation space."""
    env = data_envs.SequenceDataEnv(data_envs.copy_stream(2, n=1), 16)
    self.assertEqual(env.action_space.n, 16)
    self.assertEqual(env.observation_space.n, 16)
Test that sequence data env returns correct action/observation space.
trax/rl/envs/data_envs_test.py