mirror of
https://github.com/neuml/txtai.git
synced 2026-10-11 21:39:55 +00:00
Add comment
This commit is contained in:
@@ -62,8 +62,10 @@ class TestApp(unittest.TestCase):
|
||||
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)))
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user