Fix command_test failure

ustclug · Oct 24, 2024 · aec855b · aec855b
1 parent 35a4945
commit aec855b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/frontend/tests/command_test.py b/frontend/tests/command_test.py
@@ -105,10 +105,10 @@ def setUp(self) -> None:
             pass
         os.mkdir(self.DIR_NAME)
         # create some example README.md files
-        for i in range(3):
+        for i, file in enumerate(FILES):
             os.mkdir(f"{self.DIR_NAME}/example{i}")
             with open(f"{self.DIR_NAME}/example{i}/README.md", "w") as f:
-                f.write(FILES[i])
+                f.write(file)
 
     def tearDown(self) -> None:
         shutil.rmtree(self.DIR_NAME)