def fake_module(fake_driver_name): import sys from types import ModuleType module = ModuleType(fake_driver_name) class Dataset: """Fake Dataset""" def __init__(self, key): self.key = key module.Dataset = Dataset sys.modules[module.__name__] = module yield module del sys.modules[module.__name__]