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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
"""Tests for import management functionality."""
from __future__ import annotations
from typing import TYPE_CHECKING
import pytest
from datamodel_code_generator.imports import Import, Imports
if TYPE_CHECKING:
from collections.abc import Sequence
@pytest.mark.parametrize(
("inputs", "value"),
[
([(None, "foo")], "import foo"),
([(".", "foo")], "from . import foo"),
([("bar", "foo")], "from bar import foo"),
([("bar", "foo"), ("bar", "baz")], "from bar import baz, foo"),
([("bar", "foo"), ("rab", "oof")], "from bar import foo\nfrom rab import oof"),
([("bar", "foo"), ("bar", "foo")], "from bar import foo"),
([(None, "foo.baz")], "import foo.baz"),
],
)
def test_dump(inputs: Sequence[tuple[str | None, str]], value: str) -> None:
"""Test creating import lines."""
imports = Imports()
imports.append(Import(from_=from_, import_=import_) for from_, import_ in inputs)
assert str(imports) == value
def test_is_future_true() -> None:
"""Test that __future__ imports are identified as future imports."""
import_ = Import(from_="__future__", import_="annotations")
assert import_.is_future is True
def test_is_future_false_regular_import() -> None:
"""Test that regular imports are not identified as future imports."""
import_ = Import(from_="typing", import_="Optional")
assert import_.is_future is False
def test_is_future_false_no_from() -> None:
"""Test that imports without from_ are not identified as future imports."""
import_ = Import(from_=None, import_="os")
assert import_.is_future is False
def test_extract_future_with_future_imports() -> None:
"""Test extracting future imports from mixed imports."""
imports = Imports()
imports.append(Import(from_="__future__", import_="annotations"))
imports.append(Import(from_="typing", import_="Optional"))
future = imports.extract_future()
assert str(future) == "from __future__ import annotations"
assert str(imports) == "from typing import Optional"
assert "__future__" not in imports
def test_extract_future_no_future_imports() -> None:
"""Test extracting from imports without future imports."""
imports = Imports()
imports.append(Import(from_="typing", import_="Optional"))
future = imports.extract_future()
assert not str(future)
assert str(imports) == "from typing import Optional"
def test_extract_future_only_future_imports() -> None:
"""Test extracting when only future imports exist."""
imports = Imports()
imports.append(Import(from_="__future__", import_="annotations"))
future = imports.extract_future()
assert str(future) == "from __future__ import annotations"
assert not str(imports)
def test_extract_future_with_alias() -> None:
"""Test extracting future imports with alias (edge case)."""
imports = Imports()
imports.append(Import(from_="__future__", import_="annotations", alias="ann"))
imports.append(Import(from_="typing", import_="Optional"))
future = imports.extract_future()
assert "annotations as ann" in str(future)
assert "__future__" not in imports
assert "__future__" not in imports.alias
|
