MetaGPT

1
#!/usr/bin/env python
2
# -*- coding: utf-8 -*-
3
"""
4
@Time    : 2024/1/2 21:00
5
@Author  : alexanderwu
6
@File    : test_document.py
7
"""
8
from metagpt.config2 import config
9
from metagpt.document import Repo
10
from metagpt.logs import logger
11

12

13
def set_existing_repo(path):
14
    repo1 = Repo.from_path(path)
15
    repo1.set("doc/wtf_file.md", "wtf content")
16
    repo1.set("code/wtf_file.py", "def hello():\n    print('hello')")
17
    logger.info(repo1)  # check doc
18

19

20
def load_existing_repo(path):
21
    repo = Repo.from_path(path)
22
    logger.info(repo)
23
    logger.info(repo.eda())
24

25
    assert repo
26
    assert repo.get("doc/wtf_file.md").content == "wtf content"
27
    assert repo.get("code/wtf_file.py").content == "def hello():\n    print('hello')"
28

29

30
def test_repo_set_load():
31
    repo_path = config.workspace.path / "test_repo"
32
    set_existing_repo(repo_path)
33
    load_existing_repo(repo_path)
34