Update TestCodeFileEditAtomicFlow.py

TestCodeFileEditAtomicFlow.py

@@ -40,9 +40,3 @@ class TestCodeFileEditAtomicFlow(CodeFileEditAtomicFlow):
         assert "code" in input_data, "code is not passed to TestCodeFileEditAtomicFlow"
         assert "memory_files" in input_data, "memory_files is not passed to TestCodeFileEditAtomicFlow"
 
-    def _generate_input_to_writer(self, input_data: Dict[str, Any]):
-        code_str = input_data['code']
-        code_lib_location = input_data["memory_files"]["code_library"]
-        content_to_write = self._generate_content(code_lib_location, code_str)
-        file_location_to_write = self._generate_temp_file_location(code_lib_location)
-        return content_to_write, file_location_to_write