def test_unsupported_language(kwik_e_mart_app_path): config = get_augmentation_config(app_path=kwik_e_mart_app_path) resource_loader = ResourceLoader.create_resource_loader( kwik_e_mart_app_path) with pytest.raises(UnsupportedLanguageError): AugmentorFactory( config=config, language="de", resource_loader=resource_loader, ).create_augmentor()
def english_paraphraser(kwik_e_mart_app_path): config = get_augmentation_config(app_path=kwik_e_mart_app_path) language = "en" resource_loader = ResourceLoader.create_resource_loader( kwik_e_mart_app_path) augmentor = AugmentorFactory( config=config, language=language, resource_loader=resource_loader, ).create_augmentor() return augmentor
def multilingual_paraphraser(kwik_e_mart_app_path): config = get_augmentation_config(app_path=kwik_e_mart_app_path) config["augmentor_class"] = "MultiLingualParaphraser" language = "es" resource_loader = ResourceLoader.create_resource_loader( kwik_e_mart_app_path) augmentor = AugmentorFactory( config=config, language=language, resource_loader=resource_loader, ).create_augmentor() return augmentor
def english_paraphraser(kwik_e_mart_app_path, request): config = get_augmentation_config(app_path=kwik_e_mart_app_path) config['retain_entities'] = False language = "en" resource_loader = ResourceLoader.create_resource_loader( kwik_e_mart_app_path) request.cls.augmentor = AugmentorFactory( config=config, language=language, resource_loader=resource_loader, ).create_augmentor() yield None request.cls.augmentor = None
def test_spanish_paraphrases(kwik_e_mart_app_path, query): config = get_augmentation_config(app_path=kwik_e_mart_app_path) config["augmentor_class"] = "MultiLingualParaphraser" language = "es" resource_loader = ResourceLoader.create_resource_loader( kwik_e_mart_app_path) multilingual_paraphraser = AugmentorFactory( config=config, language=language, resource_loader=resource_loader, ).create_augmentor() paraphrases = multilingual_paraphraser.augment_queries([load_query(query)]) multilingual_paraphraser = None assert "aumentar el volumen" in paraphrases
def english_paraphraser_retain_entities(kwik_e_mart_app_path, request): config = get_augmentation_config(app_path=kwik_e_mart_app_path) language = "en" config['augmentor_class'] = "EnglishParaphraser" config['retain_entities'] = True query_factory = QueryFactory.create_query_factory( app_path=kwik_e_mart_app_path, duckling=True) resource_loader = ResourceLoader.create_resource_loader( app_path=kwik_e_mart_app_path, query_factory=query_factory) request.cls.query_factory = query_factory request.cls.augmentor = AugmentorFactory( config=config, language=language, resource_loader=resource_loader, ).create_augmentor() yield None request.cls.augmentor = None request.cls.query_factor = None
def resource_loader(query_factory): """A resource loader""" return ResourceLoader(APP_PATH, query_factory)
def resource_loader(): """A resource loader""" return ResourceLoader(app_path=None, query_factory=QUERY_FACTORY)