diff --git a/neon_minerva/cli.py b/neon_minerva/cli.py index bd98b77..f2b7fb5 100644 --- a/neon_minerva/cli.py +++ b/neon_minerva/cli.py @@ -137,9 +137,9 @@ def test_intents(skill_entrypoint, test_file, debug, padacioso): @click.argument("test_file") def test_text_inputs(lang, test_file): from neon_utils.file_utils import load_commented_file - from neon_minerva.integration.text_test_runner import TextTestRunner + from neon_minerva.integration.text import TextIntentTests prompts = load_commented_file(test_file).split('\n') click.echo(f"Testing {len(prompts)} prompts") - runner = TextTestRunner(prompts, lang) + runner = TextIntentTests(prompts, lang) results = runner.run_test() click.echo(pformat(results)) diff --git a/neon_minerva/integration/text_test_runner.py b/neon_minerva/integration/text.py similarity index 98% rename from neon_minerva/integration/text_test_runner.py rename to neon_minerva/integration/text.py index 9887dae..ccbd76e 100644 --- a/neon_minerva/integration/text_test_runner.py +++ b/neon_minerva/integration/text.py @@ -31,10 +31,11 @@ from ovos_bus_client.message import Message -class TextTestRunner: +class TextIntentTests: def __init__(self, prompts: List[str], lang: str = "en-us", bus_config: dict = None): bus_config = bus_config or dict() self.core_bus = MessageBusClient(**bus_config) + self.core_bus.run_in_thread() self.lang = lang self._prompts = prompts self._results = list()