init

AlgoriAgent/tests/logger_test.py

@@ -0,0 +1,77 @@
+# -*- coding: utf-8 -*-
+""" Unit test for logger chat"""
+import os
+import shutil
+import time
+import unittest
+
+from loguru import logger
+
+from agentscope.logging import setup_logger
+
+
+class LoggerTest(unittest.TestCase):
+    """
+    Unit test for logger.
+    """
+
+    def setUp(self) -> None:
+        """Setup for unit test."""
+        self.run_dir = "./logger_runs/"
+
+    def test_logger_chat(self) -> None:
+        """Logger chat."""
+
+        setup_logger(self.run_dir, level="INFO")
+
+        # str with "\n"
+        logger.chat("Test\nChat\n\nMessage\n\n")
+
+        # dict with "\n"
+        logger.chat(
+            {
+                "name": "Alice",
+                "content": "Hi!\n",
+                "url": "https://xxx.png",
+            },
+        )
+
+        # dict without content
+        logger.chat({"name": "Alice", "url": "https://xxx.png"})
+
+        # dict
+        logger.chat({"abc": 1})
+
+        # html labels
+        logger.chat({"name": "Bob", "content": "<div>abc</div"})
+
+        # To avoid that logging is not finished before the file is read
+        time.sleep(3)
+
+        with open(
+            os.path.join(self.run_dir, "logging.chat"),
+            "r",
+            encoding="utf-8",
+        ) as file:
+            lines = file.readlines()
+
+        ground_truth = [
+            '"Test\\nChat\\n\\nMessage\\n\\n"\n',
+            '{"name": "Alice", "content": "Hi!\\n", "url": "https://xxx.png'
+            '"}\n',
+            '{"name": "Alice", "url": "https://xxx.png"}\n',
+            '{"abc": 1}\n',
+            '{"name": "Bob", "content": "<div>abc</div"}\n',
+        ]
+
+        self.assertListEqual(lines, ground_truth)
+
+    def tearDown(self) -> None:
+        """Tear down for LoggerTest."""
+        logger.remove()
+        if os.path.exists(self.run_dir):
+            shutil.rmtree(self.run_dir)
+
+
+if __name__ == "__main__":
+    unittest.main()