def _mock_import(name, *args, **kwargs): if name == 'json': raise ImportError elif name == 'simplejson': sj = MagicMock() sj.__version__ = '3.10.0' return sj return realimport(name, *args, **kwargs)
def _mock_import(name, *args, **kwargs): if 'psycopg2' in name: m = MagicMock() m.__version__ = '2.4.1' return m return realimport(name, *args, **kwargs)