def test_from_import_records(self) -> None:
results = ImportRecordGroup.from_import_records(
[
ImportRecord(ImportString("builtins"), "str"),
ImportRecord(ImportString("typing"), "Any"),
ImportRecord(
ImportString("boto3"),
"Resource",
fallback=ImportRecord(ImportString("boto3"), "List"),
),
ImportRecord(ImportString("typing"), "Text", "string"),
ImportRecord(ImportString("mypy_boto3"), "MyClass"),
ImportRecord.empty(),
]
)
assert len(results) == 4
assert results[0] == ImportRecordGroup(
ImportString("sys"), [ImportRecord(ImportString("sys"))]
)
assert results[1] == ImportRecordGroup(
ImportString("typing"),
[
ImportRecord(ImportString("typing"), "Any"),
ImportRecord(ImportString("typing"), "Text", "string"),
],
)
assert results[2] == ImportRecordGroup(
ImportString("mypy_boto3"),
[ImportRecord(ImportString("mypy_boto3"), "MyClass"),],
)
assert results[3] == ImportRecordGroup(
ImportString("boto3"),
[
ImportRecord(
ImportString("boto3"),
"Resource",
fallback=ImportRecord(ImportString("boto3"), "List"),
),
],
)
def get_import_record(self) -> ImportRecord:
"""
Get empty import record, because constants do not require imports.
"""
return ImportRecord.empty()
def get_import_record(self) -> ImportRecord:
"""
Get import record required for using type annotation.
"""
return ImportRecord.empty()
def test_empty(self, ImportStringMock: MagicMock) -> None:
assert ImportRecord.empty().source == ImportStringMock.empty()
ImportStringMock.empty.assert_called_with()
def get_import_record(self) -> ImportRecord:
return ImportRecord.empty()
def get_import_record(self) -> ImportRecord:
"""
Not used.
"""
return ImportRecord.empty()
