chore: import upstream snapshot with attribution

test/python/testapp.py

@@ -0,0 +1,91 @@
+"""
+Application module tests
+"""
+
+import unittest
+import types
+
+from txtai.app import Application
+from txtai.pipeline import Pipeline
+
+
+class TestApp(unittest.TestCase):
+    """
+    Application tests.
+    """
+
+    def testConfig(self):
+        """
+        Test a file not found config exception
+        """
+
+        with self.assertRaises(FileNotFoundError):
+            Application.read("No file here")
+
+    def testParameter(self):
+        """
+        Test resolving application parameter
+        """
+
+        app = Application(
+            """
+            testapp.TestPipeline:
+                application:
+        """
+        )
+
+        # Check that application instance is not None
+        self.assertIsNotNone(app.pipelines["testapp.TestPipeline"].application)
+
+    def testStream(self):
+        """
+        Test workflow streams
+        """
+
+        app = Application(
+            """
+            workflow:
+                stream:
+                    stream:
+                        action: testapp.TestStream
+                    tasks:
+                        - nop
+                batchstream:
+                    stream:
+                        action: testapp.TestStream
+                        batch: True
+                    tasks:
+                        - nop
+        """
+        )
+
+        def generator():
+            yield 10
+
+        # Test single stream
+        self.assertEqual(list(app.workflow("stream", [10])), list(range(10)))
+
+        # Test batch stream
+        self.assertEqual(list(app.workflow("batchstream", generator())), list(range(10)))
+
+
+class TestPipeline(Pipeline):
+    """
+    Test pipeline with an application parameter.
+    """
+
+    def __init__(self, application):
+        self.application = application
+
+
+class TestStream:
+    """
+    Test workflow stream
+    """
+
+    def __call__(self, arg):
+        if isinstance(arg, types.GeneratorType):
+            for x in arg:
+                yield from range(int(x))
+        else:
+            yield from range(int(arg))