Update TestCodeFileEditAtomicFlow.py

TestCodeFileEditAtomicFlow.py

@@ -9,7 +9,7 @@ class TestCodeFileEditAtomicFlow(CodeFileEditAtomicFlow):
         import_code = (
             f"import sys\n"
             f"sys.path.insert(0, '{module_dir}')\n"
-            f"import {module_name}\n"
+            f"from {module_name} import *\n"
         )
         return import_code