XY

2021-05-28 02:22:59 +08:00
parent 36311809e8
commit 89448e433f
3 changed files with 9 additions and 9 deletions
--- a/tests/test_synthetic_env.py
+++ b/tests/test_synthetic_env.py
@@ -16,5 +16,5 @@ class TestSynethicEnv(unittest.TestCase):
        for version in versions:
            env = get_synthetic_env(version=version)
        print(env)
-        for timestamp, tau in env:
-            self.assertEqual(tau.shape, (1000, env.ndim))
+        for timestamp, (x, y) in env:
+            self.assertEqual(x.shape, (1000, env.ndim))