def test_import_success(): import sys from did.utils import _import s = _import("sys", True) assert s is sys s = _import("blah", True) assert s is None
def test_import_failure(): from did.utils import _import with pytest.raises(ImportError): _import("blah", False)