def test_train_named_model(component_builder):
_config = utilities.base_test_conf("keyword")
_config['name'] = "my_keyword_model"
(trained, persisted_path) = utilities.run_train(_config, component_builder)
assert trained.pipeline
# should be saved in a dir named after a project
assert os.path.basename(os.path.dirname(os.path.normpath(persisted_path))) == "my_keyword_model"
def test_train_named_model(component_builder):
_config = utilities.base_test_conf("keyword")
_config['project'] = "my_keyword_model"
(trained, persisted_path) = utilities.run_train(_config, component_builder)
assert trained.pipeline
normalized_path = os.path.dirname(os.path.normpath(persisted_path))
# should be saved in a dir named after a project
assert os.path.basename(normalized_path) == "my_keyword_model"
def test_train_model(pipeline_template, component_builder):
_config = utilities.base_test_conf(pipeline_template)
(trained, persisted_path) = utilities.run_train(_config, component_builder)
assert trained.pipeline
loaded = utilities.load_interpreter_for_model(_config, persisted_path, component_builder)
assert loaded.pipeline
assert loaded.parse("hello") is not None
assert loaded.parse("Hello today is Monday, again!") is not None
def test_train_model_multithread(component_builder):
_config = utilities.base_test_conf("all_components")
_config['num_threads'] = 2
(trained, persisted_path) = utilities.run_train(_config, component_builder)
assert trained.pipeline
loaded = utilities.load_interpreter_for_model(_config, persisted_path, component_builder)
assert loaded.pipeline
assert loaded.parse("hello") is not None
assert loaded.parse("Hello today is Monday, again!") is not None
def test_train_model_noents(component_builder):
_config = utilities.base_test_conf("all_components")
_config['data'] = "./data/test/demo-rasa-noents.json"
(trained, persisted_path) = utilities.run_train(_config, component_builder)
assert trained.pipeline
loaded = utilities.load_interpreter_for_model(_config, persisted_path, component_builder)
assert loaded.pipeline
assert loaded.parse("hello") is not None
assert loaded.parse("Hello today is Monday, again!") is not None
def test_handles_pipeline_with_non_existing_component(component_builder):
_config = utilities.base_test_conf("spacy_sklearn")
_config['pipeline'].append("my_made_up_component")
with pytest.raises(Exception) as execinfo:
utilities.run_train(_config, component_builder)
assert "Failed to find component" in str(execinfo.value)
def test_train_model_empty_pipeline(component_builder):
_config = utilities.base_test_conf(
pipeline_template=None) # Should return an empty pipeline
with pytest.raises(ValueError):
utilities.run_train(_config, component_builder)
def test_handles_pipeline_with_non_existing_component(component_builder):
_config = utilities.base_test_conf("spacy_sklearn")
_config['pipeline'].append("my_made_up_component")
with pytest.raises(Exception) as execinfo:
utilities.run_train(_config, component_builder)
assert "Failed to find component" in str(execinfo.value)
def test_train_model_empty_pipeline(component_builder):
_config = utilities.base_test_conf(pipeline_template=None) # Should return an empty pipeline
with pytest.raises(ValueError):
utilities.run_train(_config, component_builder)
