vsajip · December 16, 2022 21:26 · Jun 10, 2012 · Jun 4, 2012 · Jun 4, 2012
diff --git a/debounce.py b/debounce.py
@@ -1,13 +1,14 @@
 from threading import Timer
 
+
 def debounce(wait):
     """ Decorator that will postpone a functions
         execution until after wait seconds
         have elapsed since the last time it was invoked. """
     def decorator(fn):
-        def debounced(*args):
+        def debounced(*args, **kwargs):
             def call_it():
-                return fn(*args)
+                fn(*args, **kwargs)
             try:
                 debounced.t.cancel()
             except(AttributeError):
@@ -16,8 +17,3 @@ def call_it():
             debounced.t.start()
         return debounced
     return decorator
-
-
-@debounce(10)
-def do_expensive_thing(foo):
-    print 'My, that was expensive: %s' % foo
diff --git a/test_debounce.py b/test_debounce.py
@@ -0,0 +1,39 @@
+import unittest
+import time
+from debounce import debounce
+
+
+class TestDebounce(unittest.TestCase):
+
+    @debounce(10)
+    def increment(self):
+        """ Simple function that
+            increments a counter when
+            called, used to test the
+            debounce function decorator """
+        self.count += 1
+
+    def setUp(self):
+        self.count = 0
+
+    def test_debounce(self):
+        """ Test that the increment
+            function is being debounced.
+            The counter should only be incremented
+            once 10 seconds after the last call
+            to the function """
+        self.assertTrue(self.count == 0)
+        self.increment()
+        self.increment()
+        time.sleep(9)
+        self.assertTrue(self.count == 0)
+        self.increment()
+        self.increment()
+        self.increment()
+        self.increment()
+        self.assertTrue(self.count == 0)
+        time.sleep(10)
+        self.assertTrue(self.count == 1)
+
+if __name__ == '__main__':
+    unittest.main()
diff --git a/debounce.py b/debounce.py
@@ -1,3 +1,5 @@
+from threading import Timer
+
 def debounce(wait):
     """ Decorator that will postpone a functions
         execution until after wait seconds

diff --git a/debounce.py b/debounce.py
@@ -0,0 +1,21 @@
+def debounce(wait):
+    """ Decorator that will postpone a functions
+        execution until after wait seconds
+        have elapsed since the last time it was invoked. """
+    def decorator(fn):
+        def debounced(*args):
+            def call_it():
+                return fn(*args)
+            try:
+                debounced.t.cancel()
+            except(AttributeError):
+                pass
+            debounced.t = Timer(wait, call_it)
+            debounced.t.start()
+        return debounced
+    return decorator
+
+
+@debounce(10)
+def do_expensive_thing(foo):
+    print 'My, that was expensive: %s' % foo