From b8b94cc791466b7c06a0114deef75ce501151bac Mon Sep 17 00:00:00 2001 From: D-X-Y <280835372@qq.com> Date: Fri, 28 May 2021 02:40:08 +0800 Subject: [PATCH] XY --- tests/test_synthetic_env.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_synthetic_env.py b/tests/test_synthetic_env.py index f96d6fc..4f2d75a 100644 --- a/tests/test_synthetic_env.py +++ b/tests/test_synthetic_env.py @@ -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))