def __init__(self, fully_qualified_name: str,
schema_loader: SchemaLoader) -> None:
super().__init__(fully_qualified_name, schema_loader,
self.ATTRIBUTE_AGGREGATES)
self.version = self._spec.get(self.ATTRIBUTE_VERSION, None)
self.import_list = self._spec.get(self.ATTRIBUTE_IMPORT, [])
self.schema_context = SchemaContext(self.import_list)
def test_import_module_and_identifier_with_alias():
spec = [{'Module': 'dateutil', 'Identifiers': ['parser as my_func']}]
schema_context = SchemaContext(spec)
context = schema_context.context
assert not context.local_context
assert context.global_context
from dateutil import parser
assert context.global_context['my_func'] == parser
def test_import_just_module_with_alias():
spec = [{'Module': 'dateutil as my_mod'}]
schema_context = SchemaContext(spec)
context = schema_context.context
assert not context.local_context
assert context.global_context
import dateutil
assert context.global_context['my_mod'] == dateutil
def test_import_module_and_multiple_identifiers():
spec = [{'Module': 'dateutil', 'Identifiers': ['parser', 'tz']}]
schema_context = SchemaContext(spec)
context = schema_context.context
assert not context.local_context
assert context.global_context
from dateutil import parser
from dateutil import tz
assert context.global_context['parser'] == parser
assert context.global_context['tz'] == tz
def test_import_module_and_identifier_error():
spec = [{'Module': 'dateutil', 'Identifiers': ['unknown_func']}]
schema_context = SchemaContext(spec)
with pytest.raises(ImportError,
match='cannot import name \'unknown_func\''):
assert schema_context.context
def test_import_just_module_error():
spec = [{'Module': 'unknown_module'}]
schema_context = SchemaContext(spec)
with pytest.raises(ModuleNotFoundError,
match='No module named \'unknown_module\''):
assert schema_context.context
def test_import_empty():
schema_context = SchemaContext([])
context = schema_context.context
assert not context.local_context
assert not context.global_context