Fix pass a wrong type env object

Signed-off-by: Luyao Huang <[email protected]>
iccaszhulili · Aug 21, 2019 · 45e303a · 45e303a
1 parent b6afbc5
commit 45e303a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_case_generator.py b/tests/test_case_generator.py
@@ -79,7 +79,7 @@ def test_gen_cases_special_method():
 
     src_env = Env()
     start_env = list(Env.gen_require_env(mock_func1))
-    end_env = Env.gen_require_env(mock_func6)
+    end_env = list(Env.gen_require_env(mock_func6))[0]
     cases = list(case_generator.gen_cases_special(src_env, start_env, end_env))
     assert len(cases) == 17
     for case in cases: