diff --git a/tests/nlu_nlg/nlu/test_lexical_analyzer_parser.py b/tests/nlu_nlg/nlu/test_lexical_analyzer_parser.py index 2109098..d7102ed 100644 --- a/tests/nlu_nlg/nlu/test_lexical_analyzer_parser.py +++ b/tests/nlu_nlg/nlu/test_lexical_analyzer_parser.py @@ -6,9 +6,8 @@ from unittest.mock import patch, MagicMock from godelOS.nlu_nlg.nlu.lexical_analyzer_parser import ( - -import pytest -import importlib + LexicalAnalyzerParser, SyntacticParseOutput, Token +) class TestLexicalAnalyzerParser(unittest.TestCase): """Test cases for the LexicalAnalyzerParser class.""" @@ -336,4 +335,4 @@ def test_is_dependent_on(self): if __name__ == '__main__': - unittest.main() \ No newline at end of file + unittest.main() diff --git a/tests/nlu_nlg/nlu/test_pipeline.py b/tests/nlu_nlg/nlu/test_pipeline.py index 9659d4a..4748edb 100644 --- a/tests/nlu_nlg/nlu/test_pipeline.py +++ b/tests/nlu_nlg/nlu/test_pipeline.py @@ -26,9 +26,8 @@ LexiconOntologyLinker, Lexicon, Ontology ) from godelOS.nlu_nlg.nlu.pipeline import ( - -import pytest -import importlib + NLUPipeline, NLUResult, create_nlu_pipeline +) class TestNLUPipeline(unittest.TestCase): """Test cases for the NLUPipeline class.""" @@ -225,4 +224,4 @@ def test_end_to_end_integration(self): if __name__ == '__main__': - unittest.main() \ No newline at end of file + unittest.main()