Black

tests/components/switch/test_init.py

@@ -18,11 +18,10 @@ class TestSwitch(unittest.TestCase):
     def setUp(self):
         """Set up things to be run when tests are started."""
         self.hass = get_test_home_assistant()
-        platform = getattr(self.hass.components, 'test.switch')
+        platform = getattr(self.hass.components, "test.switch")
         platform.init()
         # Switch 1 is ON, switch 2 is OFF
-        self.switch_1, self.switch_2, self.switch_3 = \
-            platform.DEVICES
+        self.switch_1, self.switch_2, self.switch_3 = platform.DEVICES
 
     # pylint: disable=invalid-name
     def tearDown(self):
@@ -32,11 +31,10 @@ class TestSwitch(unittest.TestCase):
     def test_methods(self):
         """Test is_on, turn_on, turn_off methods."""
         assert setup_component(
-            self.hass, switch.DOMAIN, {switch.DOMAIN: {CONF_PLATFORM: 'test'}}
+            self.hass, switch.DOMAIN, {switch.DOMAIN: {CONF_PLATFORM: "test"}}
         )
         assert switch.is_on(self.hass)
-        assert STATE_ON == \
-            self.hass.states.get(switch.ENTITY_ID_ALL_SWITCHES).state
+        assert STATE_ON == self.hass.states.get(switch.ENTITY_ID_ALL_SWITCHES).state
         assert switch.is_on(self.hass, self.switch_1.entity_id)
         assert not switch.is_on(self.hass, self.switch_2.entity_id)
         assert not switch.is_on(self.hass, self.switch_3.entity_id)
@@ -56,8 +54,7 @@ class TestSwitch(unittest.TestCase):
         self.hass.block_till_done()
 
         assert not switch.is_on(self.hass)
-        assert STATE_OFF == \
-            self.hass.states.get(switch.ENTITY_ID_ALL_SWITCHES).state
+        assert STATE_OFF == self.hass.states.get(switch.ENTITY_ID_ALL_SWITCHES).state
         assert not switch.is_on(self.hass, self.switch_1.entity_id)
         assert not switch.is_on(self.hass, self.switch_2.entity_id)
         assert not switch.is_on(self.hass, self.switch_3.entity_id)
@@ -68,8 +65,7 @@ class TestSwitch(unittest.TestCase):
         self.hass.block_till_done()
 
         assert switch.is_on(self.hass)
-        assert STATE_ON == \
-            self.hass.states.get(switch.ENTITY_ID_ALL_SWITCHES).state
+        assert STATE_ON == self.hass.states.get(switch.ENTITY_ID_ALL_SWITCHES).state
         assert switch.is_on(self.hass, self.switch_1.entity_id)
         assert switch.is_on(self.hass, self.switch_2.entity_id)
         assert switch.is_on(self.hass, self.switch_3.entity_id)
@@ -77,38 +73,40 @@ class TestSwitch(unittest.TestCase):
     def test_setup_two_platforms(self):
         """Test with bad configuration."""
         # Test if switch component returns 0 switches
-        test_platform = getattr(self.hass.components, 'test.switch')
+        test_platform = getattr(self.hass.components, "test.switch")
         test_platform.init(True)
 
-        mock_entity_platform(self.hass, 'switch.test2', test_platform)
+        mock_entity_platform(self.hass, "switch.test2", test_platform)
         test_platform.init(False)
 
         assert setup_component(
-            self.hass, switch.DOMAIN, {
-                switch.DOMAIN: {CONF_PLATFORM: 'test'},
-                '{} 2'.format(switch.DOMAIN): {CONF_PLATFORM: 'test2'},
-            }
+            self.hass,
+            switch.DOMAIN,
+            {
+                switch.DOMAIN: {CONF_PLATFORM: "test"},
+                "{} 2".format(switch.DOMAIN): {CONF_PLATFORM: "test2"},
+            },
         )
 
 
 async def test_switch_context(hass, hass_admin_user):
     """Test that switch context works."""
-    assert await async_setup_component(hass, 'switch', {
-        'switch': {
-            'platform': 'test'
-        }
-    })
+    assert await async_setup_component(hass, "switch", {"switch": {"platform": "test"}})
 
     await hass.async_block_till_done()
 
-    state = hass.states.get('switch.ac')
+    state = hass.states.get("switch.ac")
     assert state is not None
 
-    await hass.services.async_call('switch', 'toggle', {
-        'entity_id': state.entity_id,
-    }, True, core.Context(user_id=hass_admin_user.id))
+    await hass.services.async_call(
+        "switch",
+        "toggle",
+        {"entity_id": state.entity_id},
+        True,
+        core.Context(user_id=hass_admin_user.id),
+    )
 
-    state2 = hass.states.get('switch.ac')
+    state2 = hass.states.get("switch.ac")
     assert state2 is not None
     assert state.state != state2.state
     assert state2.context.user_id == hass_admin_user.id