iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

XY

Browse Source
This commit is contained in:
D-X-Y 2021-05-28 02:40:08 +08:00
parent 89448e433f
commit b8b94cc791
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_synthetic_env.py
View File

@ -15,6 +15,6 @@ class TestSynethicEnv(unittest.TestCase):
versions = ["v1", "v2", "v3", "v4"]
for version in versions:
env = get_synthetic_env(version=version)
print(env)
for timestamp, (x, y) in env:
self.assertEqual(x.shape, (1000, env.ndim))
print(env)
for timestamp, (x, y) in env:
self.assertEqual(x.shape, (1000, env._data_generator.ndim))