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__]
