python-ast/test_analyzer.py at main · wunused/python-ast · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import pytest
import subprocess

@pytest.mark.parametrize("class_str, input_file, expected_file", [
    ("C", "class_hierarchy_analyzer/tests/test1/lib.py", "class_hierarchy_analyzer/tests/test1/libC.txt"),
    ("D", "class_hierarchy_analyzer/tests/test1/lib.py", "class_hierarchy_analyzer/tests/test1/libD.txt"),
    ("pinho", "class_hierarchy_analyzer/tests/test1/newApp.py", "class_hierarchy_analyzer/tests/test1/newApppinho.txt"),
    ("A", "class_hierarchy_analyzer/tests/test1/app.py", "class_hierarchy_analyzer/tests/test1/appA.txt"),
    ("B", "class_hierarchy_analyzer/tests/test1/app.py", "class_hierarchy_analyzer/tests/test1/appB.txt"),
    ("new", "class_hierarchy_analyzer/tests/test1/app.py", "class_hierarchy_analyzer/tests/test1/appnew.txt"),
    ("Animal", "class_hierarchy_analyzer/tests/test2/animal.py", "class_hierarchy_analyzer/tests/test2/animalAnimal.txt"),
    ("Animal", "class_hierarchy_analyzer/tests/test2/wild.py", "class_hierarchy_analyzer/tests/test2/wildAnimal.txt"),
    ("Feline", "class_hierarchy_analyzer/tests/test2/wild.py", "class_hierarchy_analyzer/tests/test2/wildFeline.txt"),
    ("Dog", "class_hierarchy_analyzer/tests/test2/pet.py", "class_hierarchy_analyzer/tests/test2/petDog.txt"),
    ("Cat", "class_hierarchy_analyzer/tests/test2/pet.py", "class_hierarchy_analyzer/tests/test2/petCat.txt"),
    ("A", "class_hierarchy_analyzer/tests/test3/a.py", "class_hierarchy_analyzer/tests/test3/aA.txt"),
    ("B", "class_hierarchy_analyzer/tests/test3/b.py", "class_hierarchy_analyzer/tests/test3/bB.txt"),
    ("C", "class_hierarchy_analyzer/tests/test3/c.py", "class_hierarchy_analyzer/tests/test3/cC.txt"),
    ("D", "class_hierarchy_analyzer/tests/test3/d.py", "class_hierarchy_analyzer/tests/test3/dD.txt"),
    ("Animal", "class_hierarchy_analyzer/tests/test4/hierarchy.py", "class_hierarchy_analyzer/tests/test4/hierarchyAnimal.txt"),
    ("Mammal", "class_hierarchy_analyzer/tests/test4/hierarchy.py", "class_hierarchy_analyzer/tests/test4/hierarchyMammal.txt"),
    ("Human", "class_hierarchy_analyzer/tests/test4/hierarchy.py", "class_hierarchy_analyzer/tests/test4/hierarchyHuman.txt"),
    ("FormalLanguage", "class_hierarchy_analyzer/tests/test5/overriding.py", "class_hierarchy_analyzer/tests/test5/overridingFormalLanguage.txt"),
    ("Colloquial", "class_hierarchy_analyzer/tests/test5/overriding2.py", "class_hierarchy_analyzer/tests/test5/overriding2Colloquial.txt"),
    ("Person", "class_hierarchy_analyzer/tests/test6/person.py", "class_hierarchy_analyzer/tests/test6/personPerson.txt"),
    ("Employee", "class_hierarchy_analyzer/tests/test6/employee.py", "class_hierarchy_analyzer/tests/test6/employeeEmployee.txt"),
    ("Manager", "class_hierarchy_analyzer/tests/test6/manager.py", "class_hierarchy_analyzer/tests/test6/managerManager.txt"),
    ("latin", "class_hierarchy_analyzer/tests/test7/latin.py", "class_hierarchy_analyzer/tests/test7/latinlatin.txt"),
    ("french", "class_hierarchy_analyzer/tests/test7/french.py", "class_hierarchy_analyzer/tests/test7/frenchfrench.txt"),
    ("italian", "class_hierarchy_analyzer/tests/test7/italian.py", "class_hierarchy_analyzer/tests/test7/italianitalian.txt"),
    ("english", "class_hierarchy_analyzer/tests/test7/english.py", "class_hierarchy_analyzer/tests/test7/englishenglish.txt"),
    ("UglySoup", "class_hierarchy_analyzer/tests/bs4-import/app.py", "class_hierarchy_analyzer/tests/bs4-import/appUglySoup.txt"),
    ("Question", "class_hierarchy_analyzer/tests/django-import/djangotutorial/polls/models.py", "class_hierarchy_analyzer/tests/django-import/djangoQuestion.txt"),
    ("Choice", "class_hierarchy_analyzer/tests/django-import/djangotutorial/polls/models.py", "class_hierarchy_analyzer/tests/django-import/djangoChoice.txt"),
    ("TestCategorizeByAge", "class_hierarchy_analyzer/tests/unittest-import/test_age.py", "class_hierarchy_analyzer/tests/unittest-import/testcategorize.txt"),
    ("someClass", "class_hierarchy_analyzer/tests/testForDot/app.py", "class_hierarchy_analyzer/tests/testForDot/app.txt"),
    ("someClass2", "class_hierarchy_analyzer/tests/testForDot/app2.py", "class_hierarchy_analyzer/tests/testForDot/app2.txt"),
])


def test_process(class_str, input_file, expected_file):
    result = subprocess.run(
        ["python3", "class_hierarchy_analyzer/cha.py", "-c", class_str, input_file],
        capture_output=True,
        text=True
    )

    assert result.returncode == 0, f"Script failed: {result.stderr}"


    with open(expected_file) as f:
        expected = f.read()

    assert result.stdout.strip() == expected.strip()