This commit is contained in:
D-X-Y 2021-05-28 02:40:08 +08:00
parent 89448e433f
commit b8b94cc791

View File

@ -17,4 +17,4 @@ class TestSynethicEnv(unittest.TestCase):
env = get_synthetic_env(version=version) env = get_synthetic_env(version=version)
print(env) print(env)
for timestamp, (x, y) in env: for timestamp, (x, y) in env:
self.assertEqual(x.shape, (1000, env.ndim)) self.assertEqual(x.shape, (1000, env._data_generator.ndim))