merge main to dev

tests/data/incremental_dev_project/mock.py

@@ -0,0 +1,466 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/01/17
+@Author  : mannaandpoem
+@File    : mock.py
+"""
+NEW_REQUIREMENT_SAMPLE = """
+Adding graphical interface functionality to enhance the user experience in the number-guessing game. The existing number-guessing game currently relies on command-line input for numbers. The goal is to introduce a graphical interface to improve the game's usability and visual appeal
+"""
+
+PRD_SAMPLE = """
+## Language
+
+en_us
+
+## Programming Language
+
+Python
+
+## Original Requirements
+
+Make a simple number guessing game
+
+## Product Goals
+
+- Ensure a user-friendly interface for the game
+- Provide a challenging yet enjoyable game experience
+- Design the game to be easily extendable for future features
+
+## User Stories
+
+- As a player, I want to guess numbers and receive feedback on whether my guess is too high or too low
+- As a player, I want to be able to set the difficulty level by choosing the range of possible numbers
+- As a player, I want to see my previous guesses to strategize my next guess
+- As a player, I want to know how many attempts it took me to guess the number once I get it right
+
+## Competitive Analysis
+
+- Guess The Number Game A: Basic text interface, no difficulty levels
+- Number Master B: Has difficulty levels, but cluttered interface
+- Quick Guess C: Sleek design, but lacks performance tracking
+- NumGuess D: Good performance tracking, but not mobile-friendly
+- GuessIt E: Mobile-friendly, but too many ads
+- Perfect Guess F: Offers hints, but the hints are not very helpful
+- SmartGuesser G: Has a learning mode, but lacks a competitive edge
+
+## Competitive Quadrant Chart
+
+quadrantChart
+    title "User Engagement and Game Complexity"
+    x-axis "Low Complexity" --> "High Complexity"
+    y-axis "Low Engagement" --> "High Engagement"
+    quadrant-1 "Too Simple"
+    quadrant-2 "Niche Appeal"
+    quadrant-3 "Complex & Unengaging"
+    quadrant-4 "Sweet Spot"
+    "Guess The Number Game A": [0.2, 0.4]
+    "Number Master B": [0.5, 0.3]
+    "Quick Guess C": [0.6, 0.7]
+    "NumGuess D": [0.4, 0.6]
+    "GuessIt E": [0.7, 0.5]
+    "Perfect Guess F": [0.6, 0.4]
+    "SmartGuesser G": [0.8, 0.6]
+    "Our Target Product": [0.5, 0.8]
+
+## Requirement Analysis
+
+The game should be simple yet engaging, allowing players of different skill levels to enjoy it. It should provide immediate feedback and track the player's performance. The game should also be designed with a clean and intuitive interface, and it should be easy to add new features in the future.
+
+## Requirement Pool
+
+- ['P0', 'Implement the core game logic to randomly select a number and allow the user to guess it']
+- ['P0', 'Design a user interface that displays the game status and results clearly']
+- ['P1', 'Add difficulty levels by varying the range of possible numbers']
+- ['P1', 'Keep track of and display the number of attempts for each game session']
+- ['P2', "Store and show the history of the player's guesses during a game session"]
+
+## UI Design draft
+
+The UI will feature a clean and minimalist design with a number input field, submit button, and messages area to provide feedback. There will be options to select the difficulty level and a display showing the number of attempts and history of past guesses.
+
+## Anything UNCLEAR"""
+
+DESIGN_SAMPLE = """
+## Implementation approach
+
+We will create a Python-based number guessing game with a simple command-line interface. For the user interface, we will use the built-in 'input' and 'print' functions for interaction. The random library will be used for generating random numbers. We will structure the code to be modular and easily extendable, separating the game logic from the user interface.
+
+## File list
+
+- main.py
+- game.py
+- ui.py
+
+## Data structures and interfaces
+
+
+classDiagram
+    class Game {
+        -int secret_number
+        -int min_range
+        -int max_range
+        -list attempts
+        +__init__(difficulty: str)
+        +start_game()
+        +check_guess(guess: int) str
+        +get_attempts() int
+        +get_history() list
+    }
+    class UI {
+        +start()
+        +display_message(message: str)
+        +get_user_input(prompt: str) str
+        +show_attempts(attempts: int)
+        +show_history(history: list)
+        +select_difficulty() str
+    }
+    class Main {
+        +main()
+    }
+    Main --> UI
+    UI --> Game
+
+
+## Program call flow
+
+
+sequenceDiagram
+    participant M as Main
+    participant UI as UI
+    participant G as Game
+    M->>UI: start()
+    UI->>UI: select_difficulty()
+    UI-->>G: __init__(difficulty)
+    G->>G: start_game()
+    loop Game Loop
+        UI->>UI: get_user_input("Enter your guess:")
+        UI-->>G: check_guess(guess)
+        G->>UI: display_message(feedback)
+        G->>UI: show_attempts(attempts)
+        G->>UI: show_history(history)
+    end
+    G->>UI: display_message("Correct! Game over.")
+    UI->>M: main()  # Game session ends
+
+
+## Anything UNCLEAR
+
+The requirement analysis suggests the need for a clean and intuitive interface. Since we are using a command-line interface, we need to ensure that the text-based UI is as user-friendly as possible. Further clarification on whether a graphical user interface (GUI) is expected in the future would be helpful for planning the extendability of the game."""
+
+TASKS_SAMPLE = """
+## Required Python packages
+
+- random==2.2.1
+
+## Required Other language third-party packages
+
+- No third-party dependencies required
+
+## Logic Analysis
+
+- ['game.py', 'Contains Game class with methods __init__, start_game, check_guess, get_attempts, get_history and uses random library for generating secret_number']
+- ['ui.py', 'Contains UI class with methods start, display_message, get_user_input, show_attempts, show_history, select_difficulty and interacts with Game class']
+- ['main.py', 'Contains Main class with method main that initializes UI class and starts the game loop']
+
+## Task list
+
+- game.py
+- ui.py
+- main.py
+
+## Full API spec
+
+
+
+## Shared Knowledge
+
+`game.py` contains the core game logic and is used by `ui.py` to interact with the user. `main.py` serves as the entry point to start the game.
+
+## Anything UNCLEAR
+
+The requirement analysis suggests the need for a clean and intuitive interface. Since we are using a command-line interface, we need to ensure that the text-based UI is as user-friendly as possible. Further clarification on whether a graphical user interface (GUI) is expected in the future would be helpful for planning the extendability of the game."""
+
+OLD_CODE_SAMPLE = """
+--- game.py
+```## game.py
+
+import random
+
+class Game:
+    def __init__(self, difficulty: str = 'medium'):
+        self.min_range, self.max_range = self._set_difficulty(difficulty)
+        self.secret_number = random.randint(self.min_range, self.max_range)
+        self.attempts = []
+
+    def _set_difficulty(self, difficulty: str):
+        difficulties = {
+            'easy': (1, 10),
+            'medium': (1, 100),
+            'hard': (1, 1000)
+        }
+        return difficulties.get(difficulty, (1, 100))
+
+    def start_game(self):
+        self.secret_number = random.randint(self.min_range, self.max_range)
+        self.attempts = []
+
+    def check_guess(self, guess: int) -> str:
+        self.attempts.append(guess)
+        if guess < self.secret_number:
+            return "It's higher."
+        elif guess > self.secret_number:
+            return "It's lower."
+        else:
+            return "Correct! Game over."
+
+    def get_attempts(self) -> int:
+        return len(self.attempts)
+
+    def get_history(self) -> list:
+        return self.attempts```
+
+--- ui.py
+```## ui.py
+
+from game import Game
+
+class UI:
+    def start(self):
+        difficulty = self.select_difficulty()
+        game = Game(difficulty)
+        game.start_game()
+        self.display_welcome_message(game)
+
+        feedback = ""
+        while feedback != "Correct! Game over.":
+            guess = self.get_user_input("Enter your guess: ")
+            if self.is_valid_guess(guess):
+                feedback = game.check_guess(int(guess))
+                self.display_message(feedback)
+                self.show_attempts(game.get_attempts())
+                self.show_history(game.get_history())
+            else:
+                self.display_message("Please enter a valid number.")
+
+    def display_welcome_message(self, game):
+        print("Welcome to the Number Guessing Game!")
+        print(f"Guess the number between {game.min_range} and {game.max_range}.")
+
+    def is_valid_guess(self, guess):
+        return guess.isdigit()
+
+    def display_message(self, message: str):
+        print(message)
+
+    def get_user_input(self, prompt: str) -> str:
+        return input(prompt)
+
+    def show_attempts(self, attempts: int):
+        print(f"Number of attempts: {attempts}")
+
+    def show_history(self, history: list):
+        print("Guess history:")
+        for guess in history:
+            print(guess)
+
+    def select_difficulty(self) -> str:
+        while True:
+            difficulty = input("Select difficulty (easy, medium, hard): ").lower()
+            if difficulty in ['easy', 'medium', 'hard']:
+                return difficulty
+            else:
+                self.display_message("Invalid difficulty. Please choose 'easy', 'medium', or 'hard'.")```
+
+--- main.py
+```## main.py
+
+from ui import UI
+
+class Main:
+    def main(self):
+        user_interface = UI()
+        user_interface.start()
+
+if __name__ == "__main__":
+    main_instance = Main()
+    main_instance.main()```
+"""
+
+REFINED_PRD_JSON = {
+    "Language": "en_us",
+    "Programming Language": "Python",
+    "Refined Requirements": "Adding graphical interface functionality to enhance the user experience in the number-guessing game.",
+    "Project Name": "number_guessing_game",
+    "Refined Product Goals": [
+        "Ensure a user-friendly interface for the game with the new graphical interface",
+        "Provide a challenging yet enjoyable game experience with visual enhancements",
+        "Design the game to be easily extendable for future features, including graphical elements",
+    ],
+    "Refined User Stories": [
+        "As a player, I want to interact with a graphical interface to guess numbers and receive visual feedback on my guesses",
+        "As a player, I want to easily select the difficulty level through the graphical interface",
+        "As a player, I want to visually track my previous guesses and the number of attempts in the graphical interface",
+        "As a player, I want to be congratulated with a visually appealing message when I guess the number correctly",
+    ],
+    "Competitive Analysis": [
+        "Guess The Number Game A: Basic text interface, no difficulty levels",
+        "Number Master B: Has difficulty levels, but cluttered interface",
+        "Quick Guess C: Sleek design, but lacks performance tracking",
+        "NumGuess D: Good performance tracking, but not mobile-friendly",
+        "GuessIt E: Mobile-friendly, but too many ads",
+        "Perfect Guess F: Offers hints, but the hints are not very helpful",
+        "SmartGuesser G: Has a learning mode, but lacks a competitive edge",
+        "Graphical Guess H: Graphical interface, but poor user experience due to complex design",
+    ],
+    "Competitive Quadrant Chart": 'quadrantChart\n    title "User Engagement and Game Complexity with Graphical Interface"\n    x-axis "Low Complexity" --> "High Complexity"\n    y-axis "Low Engagement" --> "High Engagement"\n    quadrant-1 "Too Simple"\n    quadrant-2 "Niche Appeal"\n    quadrant-3 "Complex & Unengaging"\n    quadrant-4 "Sweet Spot"\n    "Guess The Number Game A": [0.2, 0.4]\n    "Number Master B": [0.5, 0.3]\n    "Quick Guess C": [0.6, 0.7]\n    "NumGuess D": [0.4, 0.6]\n    "GuessIt E": [0.7, 0.5]\n    "Perfect Guess F": [0.6, 0.4]\n    "SmartGuesser G": [0.8, 0.6]\n    "Graphical Guess H": [0.7, 0.3]\n    "Our Target Product": [0.5, 0.9]',
+    "Refined Requirement Analysis": [
+        "The game should maintain its simplicity while integrating a graphical interface for enhanced engagement.",
+        "Immediate visual feedback is crucial for user satisfaction in the graphical interface.",
+        "The interface must be intuitive, allowing for easy navigation and selection of game options.",
+        "The graphical design should be clean and not detract from the game's core guessing mechanic.",
+    ],
+    "Refined Requirement Pool": [
+        ["P0", "Implement a graphical user interface (GUI) to replace the command-line interaction"],
+        [
+            "P0",
+            "Design a user interface that displays the game status, results, and feedback clearly with graphical elements",
+        ],
+        ["P1", "Incorporate interactive elements for selecting difficulty levels"],
+        ["P1", "Visualize the history of the player's guesses and the number of attempts within the game session"],
+        ["P2", "Create animations for correct or incorrect guesses to enhance user feedback"],
+        ["P2", "Ensure the GUI is responsive and compatible with various screen sizes"],
+        ["P2", "Store and show the history of the player's guesses during a game session"],
+    ],
+    "UI Design draft": "The UI will feature a modern and minimalist design with a graphical number input field, a submit button with animations, and a dedicated area for visual feedback. It will include interactive elements to select the difficulty level and a visual display for the number of attempts and history of past guesses.",
+    "Anything UNCLEAR": "",
+}
+
+REFINED_DESIGN_JSON = {
+    "Refined Implementation Approach": "To accommodate the new graphical user interface (GUI) requirements, we will leverage the Tkinter library, which is included with Python and supports the creation of a user-friendly GUI. The game logic will remain in Python, with Tkinter handling the rendering of the interface. We will ensure that the GUI is responsive and provides immediate visual feedback. The main game loop will be event-driven, responding to user inputs such as button clicks and difficulty selection.",
+    "Refined File list": ["main.py", "game.py", "ui.py", "gui.py"],
+    "Refined Data structures and interfaces": "\nclassDiagram\n    class Game {\n        -int secret_number\n        -int min_range\n        -int max_range\n        -list attempts\n        +__init__(difficulty: str)\n        +start_game()\n        +check_guess(guess: int) str\n        +get_attempts() int\n        +get_history() list\n    }\n    class UI {\n        +start()\n        +display_message(message: str)\n        +get_user_input(prompt: str) str\n        +show_attempts(attempts: int)\n        +show_history(history: list)\n        +select_difficulty() str\n    }\n    class GUI {\n        +__init__()\n        +setup_window()\n        +bind_events()\n        +update_feedback(message: str)\n        +update_attempts(attempts: int)\n        +update_history(history: list)\n        +show_difficulty_selector()\n        +animate_guess_result(correct: bool)\n    }\n    class Main {\n        +main()\n    }\n    Main --> UI\n    UI --> Game\n    UI --> GUI\n    GUI --> Game\n",
+    "Refined Program call flow": '\nsequenceDiagram\n    participant M as Main\n    participant UI as UI\n    participant G as Game\n    participant GU as GUI\n    M->>UI: start()\n    UI->>GU: setup_window()\n    GU->>GU: bind_events()\n    GU->>UI: select_difficulty()\n    UI-->>G: __init__(difficulty)\n    G->>G: start_game()\n    loop Game Loop\n        GU->>GU: show_difficulty_selector()\n        GU->>UI: get_user_input("Enter your guess:")\n        UI-->>G: check_guess(guess)\n        G->>GU: update_feedback(feedback)\n        G->>GU: update_attempts(attempts)\n        G->>GU: update_history(history)\n        GU->>GU: animate_guess_result(correct)\n    end\n    G->>GU: update_feedback("Correct! Game over.")\n    GU->>M: main()  # Game session ends\n',
+    "Anything UNCLEAR": "",
+}
+
+REFINED_TASKS_JSON = {
+    "Required Python packages": ["random==2.2.1", "Tkinter==8.6"],
+    "Required Other language third-party packages": ["No third-party dependencies required"],
+    "Refined Logic Analysis": [
+        [
+            "game.py",
+            "Contains Game class with methods __init__, start_game, check_guess, get_attempts, get_history and uses random library for generating secret_number",
+        ],
+        [
+            "ui.py",
+            "Contains UI class with methods start, display_message, get_user_input, show_attempts, show_history, select_difficulty and interacts with Game class",
+        ],
+        [
+            "gui.py",
+            "Contains GUI class with methods __init__, setup_window, bind_events, update_feedback, update_attempts, update_history, show_difficulty_selector, animate_guess_result and interacts with Game class for GUI rendering",
+        ],
+        [
+            "main.py",
+            "Contains Main class with method main that initializes UI class and starts the event-driven game loop",
+        ],
+    ],
+    "Refined Task list": ["game.py", "ui.py", "gui.py", "main.py"],
+    "Full API spec": "",
+    "Refined Shared Knowledge": "`game.py` contains the core game logic and is used by `ui.py` to interact with the user. `main.py` serves as the entry point to start the game. `gui.py` is introduced to handle the graphical user interface using Tkinter, which will interact with both `game.py` and `ui.py` for a responsive and user-friendly experience.",
+    "Anything UNCLEAR": "",
+}
+
+CODE_PLAN_AND_CHANGE_SAMPLE = {
+    "Plan": '\n1. Plan for gui.py: Develop the GUI using Tkinter to replace the command-line interface. Start by setting up the main window and event handling. Then, add widgets for displaying the game status, results, and feedback. Implement interactive elements for difficulty selection and visualize the guess history. Finally, create animations for guess feedback and ensure responsiveness across different screen sizes.\n```python\nclass GUI:\n-    pass\n+    def __init__(self):\n+        self.setup_window()\n+\n+    def setup_window(self):\n+        # Initialize the main window using Tkinter\n+        pass\n+\n+    def bind_events(self):\n+        # Bind button clicks and other events\n+        pass\n+\n+    def update_feedback(self, message: str):\n+        # Update the feedback label with the given message\n+        pass\n+\n+    def update_attempts(self, attempts: int):\n+        # Update the attempts label with the number of attempts\n+        pass\n+\n+    def update_history(self, history: list):\n+        # Update the history view with the list of past guesses\n+        pass\n+\n+    def show_difficulty_selector(self):\n+        # Show buttons or a dropdown for difficulty selection\n+        pass\n+\n+    def animate_guess_result(self, correct: bool):\n+        # Trigger an animation for correct or incorrect guesses\n+        pass\n```\n\n2. Plan for main.py: Modify the main.py to initialize the GUI and start the event-driven game loop. Ensure that the GUI is the primary interface for user interaction.\n```python\nclass Main:\n     def main(self):\n-        user_interface = UI()\n-        user_interface.start()\n+        graphical_user_interface = GUI()\n+        graphical_user_interface.setup_window()\n+        graphical_user_interface.bind_events()\n+        # Start the Tkinter main loop\n+        pass\n\n if __name__ == "__main__":\n     main_instance = Main()\n     main_instance.main()\n```\n\n3. Plan for ui.py: Refactor ui.py to work with the new GUI class. Remove command-line interactions and delegate display and input tasks to the GUI.\n```python\nclass UI:\n-    def display_message(self, message: str):\n-        print(message)\n+\n+    def display_message(self, message: str):\n+        # This method will now pass the message to the GUI to display\n+        pass\n\n-    def get_user_input(self, prompt: str) -> str:\n-        return input(prompt)\n+\n+    def get_user_input(self, prompt: str) -> str:\n+        # This method will now trigger the GUI to get user input\n+        pass\n\n-    def show_attempts(self, attempts: int):\n-        print(f"Number of attempts: {attempts}")\n+\n+    def show_attempts(self, attempts: int):\n+        # This method will now update the GUI with the number of attempts\n+        pass\n\n-    def show_history(self, history: list):\n-        print("Guess history:")\n-        for guess in history:\n-            print(guess)\n+\n+    def show_history(self, history: list):\n+        # This method will now update the GUI with the guess history\n+        pass\n```\n\n4. Plan for game.py: Ensure game.py remains mostly unchanged as it contains the core game logic. However, make minor adjustments if necessary to integrate with the new GUI.\n```python\nclass Game:\n     # No changes required for now\n```\n'
+}
+
+REFINED_CODE_INPUT_SAMPLE = """
+-----Now, game.py to be rewritten
+```## game.py
+
+import random
+
+class Game:
+    def __init__(self, difficulty: str = 'medium'):
+        self.min_range, self.max_range = self._set_difficulty(difficulty)
+        self.secret_number = random.randint(self.min_range, self.max_range)
+        self.attempts = []
+
+    def _set_difficulty(self, difficulty: str):
+        difficulties = {
+            'easy': (1, 10),
+            'medium': (1, 100),
+            'hard': (1, 1000)
+        }
+        return difficulties.get(difficulty, (1, 100))
+
+    def start_game(self):
+        self.secret_number = random.randint(self.min_range, self.max_range)
+        self.attempts = []
+
+    def check_guess(self, guess: int) -> str:
+        self.attempts.append(guess)
+        if guess < self.secret_number:
+            return "It's higher."
+        elif guess > self.secret_number:
+            return "It's lower."
+        else:
+            return "Correct! Game over."
+
+    def get_attempts(self) -> int:
+        return len(self.attempts)
+
+    def get_history(self) -> list:
+        return self.attempts```
+"""
+
+REFINED_CODE_SAMPLE = """
+## game.py
+
+import random
+
+class Game:
+    def __init__(self, difficulty: str = 'medium'):
+        # Set the difficulty level with default value 'medium'
+        self.min_range, self.max_range = self._set_difficulty(difficulty)
+        # Initialize the secret number based on the difficulty
+        self.secret_number = random.randint(self.min_range, self.max_range)
+        # Initialize the list to keep track of attempts
+        self.attempts = []
+
+    def _set_difficulty(self, difficulty: str):
+        # Define the range of numbers for each difficulty level
+        difficulties = {
+            'easy': (1, 10),
+            'medium': (1, 100),
+            'hard': (1, 1000)
+        }
+        # Return the corresponding range for the selected difficulty, default to 'medium' if not found
+        return difficulties.get(difficulty, (1, 100))
+
+    def start_game(self):
+        # Reset the secret number and attempts list for a new game
+        self.secret_number = random.randint(self.min_range, self.max_range)
+        self.attempts.clear()
+
+    def check_guess(self, guess: int) -> str:
+        # Add the guess to the attempts list
+        self.attempts.append(guess)
+        # Provide feedback based on the guess
+        if guess < self.secret_number:
+            return "It's higher."
+        elif guess > self.secret_number:
+            return "It's lower."
+        else:
+            return "Correct! Game over."
+
+    def get_attempts(self) -> int:
+        # Return the number of attempts made
+        return len(self.attempts)
+
+    def get_history(self) -> list:
+        # Return the list of attempts made
+        return self.attempts
+"""

tests/data/incremental_dev_project/readme.md

@@ -0,0 +1,3 @@
+# Code archive
+
+This folder contains a compressed package for the test_incremental_dev.py file, which is used to demonstrate the process of incremental development.

tests/metagpt/actions/test_design_api_an.py

@@ -0,0 +1,46 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/01/03
+@Author  : mannaandpoem
+@File    : test_design_api_an.py
+"""
+import pytest
+from openai._models import BaseModel
+
+from metagpt.actions.action_node import ActionNode, dict_to_markdown
+from metagpt.actions.design_api import NEW_REQ_TEMPLATE
+from metagpt.actions.design_api_an import REFINED_DESIGN_NODE
+from metagpt.llm import LLM
+from tests.data.incremental_dev_project.mock import (
+    DESIGN_SAMPLE,
+    REFINED_DESIGN_JSON,
+    REFINED_PRD_JSON,
+)
+
+
+@pytest.fixture()
+def llm():
+    return LLM()
+
+
+def mock_refined_design_json():
+    return REFINED_DESIGN_JSON
+
+
+@pytest.mark.asyncio
+async def test_write_design_an(mocker):
+    root = ActionNode.from_children(
+        "RefinedDesignAPI", [ActionNode(key="", expected_type=str, instruction="", example="")]
+    )
+    root.instruct_content = BaseModel()
+    root.instruct_content.model_dump = mock_refined_design_json
+    mocker.patch("metagpt.actions.design_api_an.REFINED_DESIGN_NODE.fill", return_value=root)
+
+    prompt = NEW_REQ_TEMPLATE.format(old_design=DESIGN_SAMPLE, context=dict_to_markdown(REFINED_PRD_JSON))
+    node = await REFINED_DESIGN_NODE.fill(prompt, llm)
+
+    assert "Refined Implementation Approach" in node.instruct_content.model_dump()
+    assert "Refined File list" in node.instruct_content.model_dump()
+    assert "Refined Data structures and interfaces" in node.instruct_content.model_dump()
+    assert "Refined Program call flow" in node.instruct_content.model_dump()

tests/metagpt/actions/test_project_management_an.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/01/03
+@Author  : mannaandpoem
+@File    : test_project_management_an.py
+"""
+import pytest
+from openai._models import BaseModel
+
+from metagpt.actions.action_node import ActionNode, dict_to_markdown
+from metagpt.actions.project_management import NEW_REQ_TEMPLATE
+from metagpt.actions.project_management_an import REFINED_PM_NODE
+from metagpt.llm import LLM
+from tests.data.incremental_dev_project.mock import (
+    REFINED_DESIGN_JSON,
+    REFINED_TASKS_JSON,
+    TASKS_SAMPLE,
+)
+
+
+@pytest.fixture()
+def llm():
+    return LLM()
+
+
+def mock_refined_tasks_json():
+    return REFINED_TASKS_JSON
+
+
+@pytest.mark.asyncio
+async def test_project_management_an(mocker):
+    root = ActionNode.from_children(
+        "RefinedProjectManagement", [ActionNode(key="", expected_type=str, instruction="", example="")]
+    )
+    root.instruct_content = BaseModel()
+    root.instruct_content.model_dump = mock_refined_tasks_json
+    mocker.patch("metagpt.actions.project_management_an.REFINED_PM_NODE.fill", return_value=root)
+
+    prompt = NEW_REQ_TEMPLATE.format(old_tasks=TASKS_SAMPLE, context=dict_to_markdown(REFINED_DESIGN_JSON))
+    node = await REFINED_PM_NODE.fill(prompt, llm)
+
+    assert "Refined Logic Analysis" in node.instruct_content.model_dump()
+    assert "Refined Task list" in node.instruct_content.model_dump()
+    assert "Refined Shared Knowledge" in node.instruct_content.model_dump()

tests/metagpt/actions/test_write_code_plan_and_change_an.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/01/03
+@Author  : mannaandpoem
+@File    : test_write_code_plan_and_change_an.py
+"""
+import pytest
+from openai._models import BaseModel
+
+from metagpt.actions.action_node import ActionNode
+from metagpt.actions.write_code import WriteCode
+from metagpt.actions.write_code_plan_and_change_an import (
+    REFINED_TEMPLATE,
+    WriteCodePlanAndChange,
+)
+from metagpt.schema import CodePlanAndChangeContext, Document
+from tests.data.incremental_dev_project.mock import (
+    CODE_PLAN_AND_CHANGE_SAMPLE,
+    DESIGN_SAMPLE,
+    NEW_REQUIREMENT_SAMPLE,
+    REFINED_CODE_INPUT_SAMPLE,
+    REFINED_CODE_SAMPLE,
+    TASKS_SAMPLE,
+)
+
+
+def mock_code_plan_and_change():
+    return CODE_PLAN_AND_CHANGE_SAMPLE
+
+
+@pytest.mark.asyncio
+async def test_write_code_plan_and_change_an(mocker):
+    root = ActionNode.from_children(
+        "WriteCodePlanAndChange", [ActionNode(key="", expected_type=str, instruction="", example="")]
+    )
+    root.instruct_content = BaseModel()
+    root.instruct_content.model_dump = mock_code_plan_and_change
+    mocker.patch("metagpt.actions.write_code_plan_and_change_an.WriteCodePlanAndChange.run", return_value=root)
+
+    requirement_doc = Document()
+    prd_docs = [Document()]
+    design_docs = [Document()]
+    tasks_docs = [Document()]
+    code_plan_and_change_context = CodePlanAndChangeContext(
+        requirement_doc=requirement_doc,
+        prd_docs=prd_docs,
+        design_docs=design_docs,
+        tasks_docs=tasks_docs,
+    )
+    node = await WriteCodePlanAndChange(context=code_plan_and_change_context).run()
+
+    assert "Plan" in node.instruct_content.model_dump()
+
+
+@pytest.mark.asyncio
+async def test_refine_code(mocker):
+    mocker.patch("metagpt.actions.write_code.WriteCodePlanAndChange.write_code", return_value=REFINED_CODE_SAMPLE)
+    prompt = REFINED_TEMPLATE.format(
+        user_requirement=NEW_REQUIREMENT_SAMPLE,
+        code_plan_and_change=CODE_PLAN_AND_CHANGE_SAMPLE,
+        design=DESIGN_SAMPLE,
+        tasks=TASKS_SAMPLE,
+        code=REFINED_CODE_INPUT_SAMPLE,
+        logs="",
+        feedback="",
+        filename="game.py",
+        summary_log="",
+    )
+    code = await WriteCode().write_code(prompt=prompt)
+    assert code
+    assert "def" in code

tests/metagpt/actions/test_write_prd_an.py

@@ -0,0 +1,49 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/01/03
+@Author  : mannaandpoem
+@File    : test_write_prd_an.py
+"""
+import pytest
+from openai._models import BaseModel
+
+from metagpt.actions.action_node import ActionNode
+from metagpt.actions.write_prd import NEW_REQ_TEMPLATE
+from metagpt.actions.write_prd_an import REFINED_PRD_NODE
+from metagpt.llm import LLM
+from tests.data.incremental_dev_project.mock import (
+    NEW_REQUIREMENT_SAMPLE,
+    PRD_SAMPLE,
+    REFINED_PRD_JSON,
+)
+
+
+@pytest.fixture()
+def llm():
+    return LLM()
+
+
+def mock_refined_prd_json():
+    return REFINED_PRD_JSON
+
+
+@pytest.mark.asyncio
+async def test_write_prd_an(mocker):
+    root = ActionNode.from_children("RefinedPRD", [ActionNode(key="", expected_type=str, instruction="", example="")])
+    root.instruct_content = BaseModel()
+    root.instruct_content.model_dump = mock_refined_prd_json
+    mocker.patch("metagpt.actions.write_prd_an.REFINED_PRD_NODE.fill", return_value=root)
+
+    prompt = NEW_REQ_TEMPLATE.format(
+        requirements=NEW_REQUIREMENT_SAMPLE,
+        old_prd=PRD_SAMPLE,
+        project_name="",
+    )
+    node = await REFINED_PRD_NODE.fill(prompt, llm)
+
+    assert "Refined Requirements" in node.instruct_content.model_dump()
+    assert "Refined Product Goals" in node.instruct_content.model_dump()
+    assert "Refined User Stories" in node.instruct_content.model_dump()
+    assert "Refined Requirement Analysis" in node.instruct_content.model_dump()
+    assert "Refined Requirement Pool" in node.instruct_content.model_dump()

tests/metagpt/test_incremental_dev.py

@@ -0,0 +1,171 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/01/03
+@Author  : mannaandpoem
+@File    : test_incremental_dev.py
+"""
+import os
+import subprocess
+import time
+
+import pytest
+from typer.testing import CliRunner
+
+from metagpt.const import TEST_DATA_PATH
+from metagpt.logs import logger
+from metagpt.startup import app
+
+runner = CliRunner()
+
+IDEAS = [
+    "Add subtraction, multiplication and division operations to the calculator. The current calculator can only perform basic addition operations, and it is necessary to introduce subtraction, multiplication, division operation into the calculator",
+    "Adding graphical interface functionality to enhance the user experience in the number-guessing game. The existing number-guessing game currently relies on command-line input for numbers. The goal is to introduce a graphical interface to improve the game's usability and visual appeal",
+    "Add a feature to remove deprecated words from the word cloud. The current word cloud generator does not support removing deprecated words. Now, The word cloud generator should support removing deprecated words. Customize deactivated words to exclude them from word cloud. Let users see all the words in the text file, and allow users to select the words they want to remove.",
+    "Add an AI opponent with fixed difficulty levels. Currently, the game only allows players to compete against themselves. Implement an AI algorithm that can playing with player. This will provide a more engaging and challenging experience for players.",
+    "Add functionality to view the history of scores. The original dice rolling game could only display the current game result, but the new requirement allows players to view the history of scores",
+    "Add functionality to view the history of scores and perform statistical analysis on them. The original dice rolling game could only display the current game result, but the new requirement allows players to view the history of scores and display the statistical analysis results of the current score",
+    "Changed score target for 2048 game from 2048 to 4096. Please change the game's score target from 2048 to 4096, and change the interface size from 4*4 to 8*8",
+    "Display the history score of the player in the 2048 game. Add a record board that can display players' historical score records so that players can trace their scores",
+    "Incremental Idea Gradually increase the speed of the snake as the game progresses. In the current version of the game, the snake’s speed remains constant throughout the gameplay. Implement a feature where the snake’s speed gradually increases over time, making the game more challenging and intense as the player progresses.",
+    "Introduce power-ups and obstacles to the game. The current version of the game only involves eating food and growing the snake. Add new elements such as power-ups that can enhance the snake’s speed or make it invincible for a short duration. At the same time, introduce obstacles like walls or enemies that the snake must avoid or overcome to continue growing.",
+]
+
+PROJECT_NAMES = [
+    "simple_add_calculator",
+    "number_guessing_game",
+    "word_cloud",
+    "Gomoku",
+    "dice_simulator_new",
+    "dice_simulator_new",
+    "pygame_2048",
+    "pygame_2048",
+    "snake_game",
+    "snake_game",
+]
+
+
+def test_simple_add_calculator():
+    result = get_incremental_dev_result(IDEAS[0], PROJECT_NAMES[0])
+    log_and_check_result(result)
+
+
+def test_number_guessing_game():
+    result = get_incremental_dev_result(IDEAS[1], PROJECT_NAMES[1])
+    log_and_check_result(result)
+
+
+def test_word_cloud():
+    result = get_incremental_dev_result(IDEAS[2], PROJECT_NAMES[2])
+    log_and_check_result(result)
+
+
+def test_gomoku():
+    result = get_incremental_dev_result(IDEAS[3], PROJECT_NAMES[3])
+    log_and_check_result(result)
+
+
+def test_dice_simulator_new():
+    for i, (idea, project_name) in enumerate(zip(IDEAS[4:6], PROJECT_NAMES[4:6]), start=1):
+        result = get_incremental_dev_result(idea, project_name)
+        log_and_check_result(result, "refine_" + str(i))
+
+
+def test_refined_pygame_2048():
+    for i, (idea, project_name) in enumerate(zip(IDEAS[6:8], PROJECT_NAMES[6:8]), start=1):
+        result = get_incremental_dev_result(idea, project_name)
+        log_and_check_result(result, "refine_" + str(i))
+
+
+def test_refined_snake_game():
+    for i, (idea, project_name) in enumerate(zip(IDEAS[8:10], PROJECT_NAMES[8:10]), start=1):
+        result = get_incremental_dev_result(idea, project_name)
+        log_and_check_result(result, "refine_" + str(i))
+
+
+def log_and_check_result(result, tag_name="refine"):
+    logger.info(result)
+    logger.info(result.output)
+    if "Aborting" in result.output:
+        assert False
+    else:
+        # After running, there will be new commit
+        cur_tag = subprocess.run(["git", "describe", "--tags"], capture_output=True, text=True).stdout.strip()
+        if cur_tag == "base":
+            assert False
+        else:
+            assert True
+            if subprocess.run(["git", "show-ref", "--verify", "--quiet", f"refs/tags/{tag_name}"]).returncode == 0:
+                tag_name += str(int(time.time()))
+            try:
+                subprocess.run(["git", "tag", tag_name], check=True)
+            except subprocess.CalledProcessError as e:
+                raise e
+
+
+def get_incremental_dev_result(idea, project_name, use_review=True):
+    project_path = TEST_DATA_PATH / "incremental_dev_project" / project_name
+    if project_path.exists():
+        raise Exception(f"Project {project_name} not exists")
+    check_or_create_base_tag(project_path)
+    args = [idea, "--inc", "--project-path", project_path]
+    if not use_review:
+        args.append("--no-code-review")
+    result = runner.invoke(app, args)
+    return result
+
+
+def check_or_create_base_tag(project_path):
+    # Change the current working directory to the specified project path
+    os.chdir(project_path)
+
+    # Initialize a Git repository
+    subprocess.run(["git", "init"], check=True)
+
+    # Check if the 'base' tag exists
+    check_base_tag_cmd = ["git", "show-ref", "--verify", "--quiet", "refs/tags/base"]
+    if subprocess.run(check_base_tag_cmd).returncode == 0:
+        has_base_tag = True
+    else:
+        has_base_tag = False
+
+    if has_base_tag:
+        logger.info("Base tag exists")
+        # Switch to the 'base' branch if it exists
+        try:
+            status = subprocess.run(["git", "status", "-s"], capture_output=True, text=True).stdout.strip()
+            if status:
+                subprocess.run(["git", "clean", "-df"])
+            subprocess.run(["git", "checkout", "-f", "base"], check=True)
+            logger.info("Switched to base branch")
+        except Exception as e:
+            logger.error("Failed to switch to base branch")
+            raise e
+
+    else:
+        logger.info("Base tag doesn't exist.")
+        # Add and commit the current code if 'base' tag doesn't exist
+        add_cmd = ["git", "add", "."]
+        commit_cmd = ["git", "commit", "-m", "Initial commit"]
+        try:
+            subprocess.run(add_cmd, check=True)
+            subprocess.run(commit_cmd, check=True)
+            logger.info("Added and committed all files with the message 'Initial commit'.")
+        except Exception as e:
+            logger.error("Failed to add and commit all files.")
+            raise e
+
+        # Add 'base' tag
+        add_base_tag_cmd = ["git", "tag", "base"]
+
+        # Check if the 'git tag' command was successful
+        try:
+            subprocess.run(add_base_tag_cmd, check=True)
+            logger.info("Added 'base' tag.")
+        except Exception as e:
+            logger.error("Failed to add 'base' tag.")
+            raise e
+
+
+if __name__ == "__main__":
+    pytest.main([__file__, "-s"])