type lifecycle hooks and extend test

fork-tongue · Nov 16, 2023 · ab07c4d · ab07c4d
1 parent c6647ad
commit ab07c4d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/observ/watcher.py b/observ/watcher.py
@@ -129,8 +129,8 @@ class Watcher:
         "_number_of_callback_args",
         "__weakref__",
     ]
-    on_created = None
-    on_destroyed = None
+    on_created: Optional[Callable[[Watcher], None]] = None
+    on_destroyed: Optional[Callable[[Watcher], None]] = None
 
     def __init__(
         self,

diff --git a/tests/test_watcher_hooks.py b/tests/test_watcher_hooks.py
@@ -25,6 +25,7 @@ def _callback():
             called += 1
 
         watch(lambda: len(a), _callback, sync=True)
+        assert len(watchers) == 1
         assert called == 0
         a.append(3)
         assert called == 1