python-ast/test_docstring.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
import pytest
import subprocess

@pytest.mark.parametrize("format_type, input_file, expected_file", [
    ("epytext", "docstring_parser/tests/epytext_cases/epytext.py", "docstring_parser/tests/epytext_cases/epytext.txt"),
    ("google", "docstring_parser/tests/google_cases/add.py", "docstring_parser/tests/google_cases/add.txt"),
    ("google", "docstring_parser/tests/google_cases/foo.py", "docstring_parser/tests/google_cases/foo.txt"),
    ("google", "docstring_parser/tests/google_cases/google.py", "docstring_parser/tests/google_cases/google.txt"),
    ("google", "docstring_parser/tests/google_cases/multiline.py", "docstring_parser/tests/google_cases/multiline.txt"),
    ("google", "docstring_parser/tests/google_cases/gg_vehicles.py", "docstring_parser/tests/google_cases/gg_vehicles.txt"),
    ("google", "docstring_parser/tests/google_cases/classann.py", "docstring_parser/tests/google_cases/classann.txt"),
    ("numpy", "docstring_parser/tests/numpy_cases/numpy.py", "docstring_parser/tests/numpy_cases/numpy.txt"),
    ("numpy", "docstring_parser/tests/numpy_cases/np_vehicles.py", "docstring_parser/tests/numpy_cases/np_vehicles.txt"),
    ("google", "docstring_parser/tests/other_cases/noannotations.py", "docstring_parser/tests/other_cases/noannotations.txt"),
    ("google", "docstring_parser/tests/other_cases/oneline.py", "docstring_parser/tests/other_cases/oneline.txt"),
    ("sphinx", "docstring_parser/tests/sphinx_cases/sphinx.py", "docstring_parser/tests/sphinx_cases/sphinx.txt"),
])

def test_process(format_type, input_file, expected_file):
    result = subprocess.run(
        ["python3", "docstring_parser/dsp.py", input_file, format_type],
        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()