Fix broken unit test

Signed-off-by: Luyao Huang <[email protected]>
iccaszhulili · Aug 20, 2019 · e0ba544 · e0ba544
1 parent a2e5c1f
commit e0ba544
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_case_generator.py b/tests/test_case_generator.py
@@ -78,8 +78,8 @@ def test_gen_cases_special_method():
     case_generator.gen_depend_map(test_funcs)
 
     src_env = Env()
-    start_env = Env.gen_require_env(mock_func1)
-    end_env = Env.gen_require_env(mock_func6)
+    start_env = list(Env.gen_require_env(mock_func1))
+    end_env = list(Env.gen_require_env(mock_func6))
     cases = list(case_generator.gen_cases_special(src_env, start_env, end_env))
     assert len(cases) == 17
     for case in cases: