add unittest for save dir

modelscope · May 28, 2024 · 4603f07 · 4603f07
1 parent 20f4b4f
commit 4603f07
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/rpc_agent_test.py b/tests/rpc_agent_test.py
@@ -4,6 +4,7 @@
 """
 import unittest
 import time
+import os
 import shutil
 from loguru import logger
 
@@ -159,14 +160,15 @@ def setUp(self) -> None:
         agentscope.init(
             project="test",
             name="rpc_agent",
-            save_dir="./test_runs",
+            save_dir="./.unittest_runs",
             save_log=True,
         )
+        self.assertTrue(os.path.exists("./.unittest_runs"))
 
     def tearDown(self) -> None:
         MonitorFactory._instance = None  # pylint: disable=W0212
         logger.remove()
-        shutil.rmtree("./test_runs")
+        shutil.rmtree("./.unittest_runs")
 
     def test_single_rpc_agent_server(self) -> None:
         """test setup a single rpc agent"""