AutoGPT

1
from agbenchmark.utils.dependencies.graphs import get_roots
2

3

4
def test_get_roots():
5
    graph = {
6
        "nodes": [
7
            {"id": "A", "data": {"category": []}},
8
            {"id": "B", "data": {"category": []}},
9
            {"id": "C", "data": {"category": []}},
10
            {"id": "D", "data": {"category": []}},
11
        ],
12
        "edges": [
13
            {"from": "A", "to": "B"},
14
            {"from": "B", "to": "C"},
15
        ],
16
    }
17

18
    result = get_roots(graph)
19
    assert set(result) == {
20
        "A",
21
        "D",
22
    }, f"Expected roots to be 'A' and 'D', but got {result}"
23

24

25
def test_no_roots():
26
    fully_connected_graph = {
27
        "nodes": [
28
            {"id": "A", "data": {"category": []}},
29
            {"id": "B", "data": {"category": []}},
30
            {"id": "C", "data": {"category": []}},
31
        ],
32
        "edges": [
33
            {"from": "A", "to": "B"},
34
            {"from": "B", "to": "C"},
35
            {"from": "C", "to": "A"},
36
        ],
37
    }
38

39
    result = get_roots(fully_connected_graph)
40
    assert not result, "Expected no roots, but found some"
41

42

43
# def test_no_rcoots():
44
#     fully_connected_graph = {
45
#         "nodes": [
46
#             {"id": "A", "data": {"category": []}},
47
#             {"id": "B", "data": {"category": []}},
48
#             {"id": "C", "data": {"category": []}},
49
#         ],
50
#         "edges": [
51
#             {"from": "A", "to": "B"},
52
#             {"from": "D", "to": "C"},
53
#         ],
54
#     }
55
#
56
#     result = get_roots(fully_connected_graph)
57
#     assert set(result) == {"A"}, f"Expected roots to be 'A', but got {result}"
58