diff --git a/tests/components/light/test_rfxtrx.py b/tests/components/light/test_rfxtrx.py
index 3eeb06be24e..cd4cecd8505 100644
--- a/tests/components/light/test_rfxtrx.py
+++ b/tests/components/light/test_rfxtrx.py
@@ -148,6 +148,32 @@ class TestLightRfxtrx(unittest.TestCase):
         self.assertTrue(entity.is_on)
         self.assertEqual(entity.brightness, 255)
 
+        entity.turn_off()
+        entity_id = rfxtrx_core.RFX_DEVICES['213c7f216'].entity_id
+        entity_hass = self.hass.states.get(entity_id)
+        self.assertEqual('Test', entity_hass.name)
+        self.assertEqual('off', entity_hass.state)
+
+        entity.turn_on()
+        entity_hass = self.hass.states.get(entity_id)
+        self.assertEqual('on', entity_hass.state)
+
+        entity.turn_off()
+        entity_hass = self.hass.states.get(entity_id)
+        self.assertEqual('off', entity_hass.state)
+
+        entity.turn_on(brightness=100)
+        entity_hass = self.hass.states.get(entity_id)
+        self.assertEqual('on', entity_hass.state)
+
+        entity.turn_on(brightness=10)
+        entity_hass = self.hass.states.get(entity_id)
+        self.assertEqual('on', entity_hass.state)
+
+        entity.turn_on(brightness=255)
+        entity_hass = self.hass.states.get(entity_id)
+        self.assertEqual('on', entity_hass.state)
+
     def test_several_lights(self):
         """Test with 3 lights."""
         self.assertTrue(_setup_component(self.hass, 'light', {
diff --git a/tests/components/sensor/test_rfxtrx.py b/tests/components/sensor/test_rfxtrx.py
index 8d5a15ec80a..2d38de5eab5 100644
--- a/tests/components/sensor/test_rfxtrx.py
+++ b/tests/components/sensor/test_rfxtrx.py
@@ -75,8 +75,9 @@ class TestSensorRfxtrx(unittest.TestCase):
         self.assertEqual(TEMP_CELSIUS, entity.unit_of_measurement)
         self.assertEqual(None, entity.state)
 
-        entity_id = rfxtrx_core.RFX_DEVICES['sensor_0502']['Temperature'].entity_id
-        entity = self.hass.states.get(entity_id)   
+        entity_id = rfxtrx_core.RFX_DEVICES['sensor_0502']['Temperature']\
+            .entity_id
+        entity = self.hass.states.get(entity_id)
         self.assertEqual('Test', entity.name)
         self.assertEqual('unknown', entity.state)
 
diff --git a/tests/components/switch/test_rfxtrx.py b/tests/components/switch/test_rfxtrx.py
index 7f4efd10c55..f0146719e75 100644
--- a/tests/components/switch/test_rfxtrx.py
+++ b/tests/components/switch/test_rfxtrx.py
@@ -155,11 +155,11 @@ class TestSwitchRfxtrx(unittest.TestCase):
         self.assertFalse(entity.is_on)
 
         entity_id = rfxtrx_core.RFX_DEVICES['213c7f216'].entity_id
-        entity_hass = self.hass.states.get(entity_id)   
+        entity_hass = self.hass.states.get(entity_id)
         self.assertEqual('Test', entity_hass.name)
         self.assertEqual('off', entity_hass.state)
         entity.turn_on()
-        entity_hass = self.hass.states.get(entity_id)   
+        entity_hass = self.hass.states.get(entity_id)
         self.assertEqual('on', entity_hass.state)
         entity.turn_off()
         entity_hass = self.hass.states.get(entity_id)