finish docstrings for aggregators

This commit is contained in:
Moishe Lettvin
2024-03-07 15:16:23 -05:00
parent 8fb92e3fd7
commit b14f08a7d5
2 changed files with 53 additions and 3 deletions

View File

@@ -1,4 +1,5 @@
import asyncio
import doctest
import functools
import unittest
@@ -118,3 +119,10 @@ class TestDailyFrameAggregators(unittest.IsolatedAsyncioTestCase):
while not sink.empty():
frame = await sink.get()
self.assertEqual(frame, expected_output_frames.pop(0))
def load_tests(loader, tests, ignore):
""" Run doctests on the aggregators module. """
from dailyai.pipeline import aggregators
tests.addTests(doctest.DocTestSuite(aggregators))
return tests