def get_engine(template_type): if template_type == DEFAULT_TEMPLATE_TYPE: return Engine else: try: external_engine = load_external_engine(template_type) except ImportError: raise NotImplementedError("No such template support") return external_engine.get_engine(template_type)
def test_load_external_engine(): load_external_engine("unknown_template_type")
def test_load_mako_engine(): # fake mako module = load_external_engine("mako") engine_class = module.get_engine("mako") assert engine_class == Engine