Diff of /test/python/util.py [000000] .. [ca4dac]

Switch to side-by-side view

--- a
+++ b/test/python/util.py
@@ -0,0 +1,28 @@
+import unittest
+import sys
+from zensols.cli import CliHarness, ApplicationFailure
+from zensols.config import ConfigFactory
+from zensols.mednlp import Application, ApplicationFactory, surpress_warnings
+
+
+class TestBase(unittest.TestCase):
+    def setUp(self):
+        import warnings
+        warnings.simplefilter("ignore", ResourceWarning)
+        surpress_warnings()
+        self.text_1 = 'He was diagnosed with kidney failure and heart disease.'
+        self.text_2 = 'He loved to smoke but Marlboro cigarettes gave John Smith lung cancer while he was in Chicago.'
+        self.maxDiff = sys.maxsize
+
+    def _get_doc_parser(self, config: str = 'default', section: str = None):
+        harness: CliHarness = ApplicationFactory.create_harness()
+        args: str = f'--config test-resources/config/{config}.conf --level=err'
+        if section is None:
+            app: Application = harness.get_instance(f'show _ {args}')
+            if isinstance(app, ApplicationFailure):
+                raise app.exception
+            return app.doc_parser
+        else:
+            harness: CliHarness = ApplicationFactory.create_harness()
+            fac: ConfigFactory = harness.get_config_factory(args)
+            return fac(section)