Add child class for single callable (restored)

pyproject.toml

@@ -39,7 +39,6 @@ dependencies = [
 [project.optional-dependencies]
 test = [
     "pytest",
-    "pytest-asyncio",
     "pytest-cov"
 ]
 

src/tqdm_publisher/__init__.py

@@ -1,3 +1,4 @@
 from ._publisher import TQDMPublisher
+from ._subscriber import TQDMProgressSubscriber
 
-__all__ = ["TQDMPublisher"]
+__all__ = ["TQDMPublisher", "TQDMProgressSubscriber"]

src/tqdm_publisher/_subscriber.py

@@ -0,0 +1,7 @@
+from ._publisher import TQDMPublisher
+
+
+class TQDMProgressSubscriber(TQDMPublisher):
+    def __init__(self, iterable, on_progress_update: callable, **tqdm_kwargs):
+        super().__init__(iterable, **tqdm_kwargs)
+        self.subscribe(lambda format_dict: on_progress_update(dict(progress_bar_id=self.id, format_dict=format_dict)))

tests/test_basic.py → tests/test_publisher.py

@@ -1,9 +1,6 @@
-import asyncio
-
-import pytest
+import time
 
 from tqdm_publisher import TQDMPublisher
-from tqdm_publisher.testing import create_tasks
 
 
 def test_initialization():
@@ -12,8 +9,7 @@ def test_initialization():
 
 
 # Test concurrent callback execution
-@pytest.mark.asyncio
-async def test_subscription_and_callback_execution():
+def test_subscription_and_callback_execution():
     n_callback_executions = dict()
 
     def test_callback(identifier, data):
@@ -26,8 +22,9 @@ def test_callback(identifier, data):
 
         assert "n" in data and "total" in data
 
-    tasks = create_tasks()
-    publisher = TQDMPublisher(asyncio.as_completed(tasks), total=len(tasks))
+    all_task_durations_in_seconds = [0.1 for _ in range(10)]
+
+    publisher = TQDMPublisher(all_task_durations_in_seconds)
 
     n_subscriptions = 10
     for i in range(n_subscriptions):
@@ -37,8 +34,8 @@ def test_callback(identifier, data):
         assert callback_id in publisher.callbacks
 
     # Simulate an update to trigger the callback
-    for f in publisher:
-        await f
+    for duration in publisher:
+        time.sleep(duration)
 
     assert len(n_callback_executions) == n_subscriptions
 
@@ -51,7 +48,7 @@ def dummy_callback(data):
         pass
 
     tasks = []
-    publisher = TQDMPublisher(asyncio.as_completed(tasks), total=len(tasks))
+    publisher = TQDMPublisher(tasks)
     callback_id = publisher.subscribe(dummy_callback)
     result = publisher.unsubscribe(callback_id)
     assert result == True

tests/test_subscriber.py

@@ -0,0 +1,32 @@
+import time
+
+from tqdm_publisher import TQDMProgressSubscriber
+
+
+def test_initialization():
+    publisher = TQDMProgressSubscriber([], on_progress_update=lambda data: print(data))
+    assert len(publisher.callbacks) == 1
+
+
+# Test single callback execution
+def test_single_callback_execution():
+    n_tasks = 100
+    n_callback_executions = 0
+
+    def test_callback(data):
+        nonlocal n_callback_executions
+        n_callback_executions += 1
+        print("CAllin")
+        assert "progress_bar_id" in data and "format_dict" in data
+        assert "n" in data["format_dict"] and "total" in data["format_dict"]
+
+    all_task_durations_in_seconds = [0.1 for _ in range(10)]
+    publisher = TQDMProgressSubscriber(
+        all_task_durations_in_seconds, on_progress_update=lambda data: test_callback(data)
+    )
+
+    # Simulate an update to trigger the callback
+    for duration in publisher:
+        time.sleep(duration)
+
+    assert n_callback_executions == len(all_task_durations_in_seconds)