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()