feat: merge main

tests/metagpt/actions/test_write_docstring.py

@@ -0,0 +1,32 @@
+import pytest
+
+from metagpt.actions.write_docstring import WriteDocstring
+
+code = '''
+def add_numbers(a: int, b: int):
+    return a + b
+
+
+class Person:
+    def __init__(self, name: str, age: int):
+        self.name = name
+        self.age = age
+
+    def greet(self):
+        return f"Hello, my name is {self.name} and I am {self.age} years old."
+'''
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    ("style", "part"),
+    [
+        ("google", "Args:"),
+        ("numpy", "Parameters"),
+        ("sphinx", ":param name:"),
+    ],
+    ids=["google", "numpy", "sphinx"]
+)
+async def test_write_docstring(style: str, part: str):
+    ret = await WriteDocstring().run(code, style=style)
+    assert part in ret

tests/metagpt/utils/test_output_parser.py

@@ -19,7 +19,7 @@ def test_parse_blocks():
 
 
 def test_parse_code():
-    test_text = "```python\nprint('Hello, world!')\n```"
+    test_text = "```python\nprint('Hello, world!')```"
     expected_result = "print('Hello, world!')"
     assert OutputParser.parse_code(test_text, 'python') == expected_result
 
@@ -27,6 +27,22 @@ def test_parse_code():
         OutputParser.parse_code(test_text, 'java')
 
 
+def test_parse_python_code():
+    expected_result = "print('Hello, world!')"
+    assert OutputParser.parse_python_code("```python\nprint('Hello, world!')```") == expected_result
+    assert OutputParser.parse_python_code("```python\nprint('Hello, world!')") == expected_result
+    assert OutputParser.parse_python_code("print('Hello, world!')") == expected_result
+    assert OutputParser.parse_python_code("print('Hello, world!')```") == expected_result
+    assert OutputParser.parse_python_code("print('Hello, world!')```") == expected_result
+    expected_result = "print('```Hello, world!```')"
+    assert OutputParser.parse_python_code("```python\nprint('```Hello, world!```')```") == expected_result
+    assert OutputParser.parse_python_code("The code is: ```python\nprint('```Hello, world!```')```") == expected_result
+    assert OutputParser.parse_python_code("xxx.\n```python\nprint('```Hello, world!```')```\nxxx") == expected_result
+
+    with pytest.raises(ValueError):
+        OutputParser.parse_python_code("xxx =")
+
+
 def test_parse_str():
     test_text = "name = 'Alice'"
     expected_result = 'Alice'

tests/metagpt/utils/test_pycst.py

@@ -0,0 +1,136 @@
+from metagpt.utils import pycst
+
+code = '''
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+from typing import overload
+
+@overload
+def add_numbers(a: int, b: int):
+    ...
+
+@overload
+def add_numbers(a: float, b: float):
+    ...
+
+def add_numbers(a: int, b: int):
+    return a + b
+
+
+class Person:
+    def __init__(self, name: str, age: int):
+        self.name = name
+        self.age = age
+
+    def greet(self):
+        return f"Hello, my name is {self.name} and I am {self.age} years old."
+'''
+
+documented_code = '''
+"""
+This is an example module containing a function and a class definition.
+"""
+
+
+def add_numbers(a: int, b: int):
+    """This function is used to add two numbers and return the result.
+
+    Parameters:
+        a: The first integer.
+        b: The second integer.
+
+    Returns:
+        int: The sum of the two numbers.
+    """
+    return a + b
+
+class Person:
+    """This class represents a person's information, including name and age.
+
+    Attributes:
+        name: The person's name.
+        age: The person's age.
+    """
+
+    def __init__(self, name: str, age: int):
+        """Creates a new instance of the Person class.
+
+        Parameters:
+            name: The person's name.
+            age: The person's age.
+        """
+        ...
+
+    def greet(self):
+        """
+        Returns a greeting message including the name and age.
+
+        Returns:
+            str: The greeting message.
+        """
+        ...
+'''
+
+
+merged_code = '''
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+This is an example module containing a function and a class definition.
+"""
+
+from typing import overload
+
+@overload
+def add_numbers(a: int, b: int):
+    ...
+
+@overload
+def add_numbers(a: float, b: float):
+    ...
+
+def add_numbers(a: int, b: int):
+    """This function is used to add two numbers and return the result.
+
+    Parameters:
+        a: The first integer.
+        b: The second integer.
+
+    Returns:
+        int: The sum of the two numbers.
+    """
+    return a + b
+
+
+class Person:
+    """This class represents a person's information, including name and age.
+
+    Attributes:
+        name: The person's name.
+        age: The person's age.
+    """
+    def __init__(self, name: str, age: int):
+        """Creates a new instance of the Person class.
+
+        Parameters:
+            name: The person's name.
+            age: The person's age.
+        """
+        self.name = name
+        self.age = age
+
+    def greet(self):
+        """
+        Returns a greeting message including the name and age.
+
+        Returns:
+            str: The greeting message.
+        """
+        return f"Hello, my name is {self.name} and I am {self.age} years old."
+'''
+
+
+def test_merge_docstring():
+    data = pycst.merge_docstring(code, documented_code)
+    print(data)
+    assert data == merged_code