Clean up tests (#4209)

tests/components/notify/test_demo.py

@@ -1,6 +1,7 @@
 """The tests for the notify demo platform."""
 import unittest
 
+from homeassistant.core import callback
 from homeassistant.bootstrap import setup_component
 import homeassistant.components.notify as notify
 from homeassistant.components.notify import demo
@@ -23,6 +24,7 @@ class TestNotifyDemo(unittest.TestCase):
         self.events = []
         self.calls = []
 
+        @callback
         def record_event(event):
             """Record event to send notification."""
             self.events.append(event)
@@ -33,6 +35,7 @@ class TestNotifyDemo(unittest.TestCase):
         """"Stop down everything that was started."""
         self.hass.stop()
 
+    @callback
     def record_calls(self, *args):
         """Helper for recording calls."""
         self.calls.append(args)