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
55
56
57
58
59
60
61
62
63
64
|
import types
import pytest
from pdoc import doc_ast
from pdoc.doc_ast import _dedent
from pdoc.doc_ast import _parse
from pdoc.doc_ast import type_checking_sections
def test_dedent():
# not indented
assert _dedent("def foo(): pass") == "def foo(): pass"
# indented
assert (
_dedent(
"""\
def foo():
pass"""
)
== "def foo():\n pass"
)
# with decorator
assert (
_dedent(
"""\
@foo
def foo():
pass"""
)
== "@foo\ndef foo():\n pass"
)
# with decorator and comment
assert (
_dedent(
"""\
@foo
# hello world
def foo():
pass"""
)
== "@foo\n# hello world\ndef foo():\n pass"
)
def test_parse_error():
with pytest.warns(UserWarning, match="Error parsing source code"):
assert _parse("!!!")
# fmt: off
@pytest.mark.parametrize("code,statements", [
("""import typing\nif typing.TYPE_CHECKING:\n\tprint(42)""", 1),
("""from typing import TYPE_CHECKING\nif TYPE_CHECKING:\n\tprint(42)\n\tprint(43)""", 2),
("""print(1234)""", 0),
])
# fmt: on
def test_type_checking_sections(code, statements, monkeypatch):
monkeypatch.setattr(doc_ast, "get_source", lambda _: code)
mod = types.ModuleType("test_type_checking_sections")
assert len(type_checking_sections(mod).body) == statements
|
