Black

tests/components/wsdot/test_sensor.py

@@ -8,8 +8,15 @@ from tests.common import get_test_home_assistant, load_fixture
 
 import homeassistant.components.wsdot.sensor as wsdot
 from homeassistant.components.wsdot.sensor import (
-    ATTR_DESCRIPTION, ATTR_TIME_UPDATED, CONF_API_KEY, CONF_ID, CONF_NAME,
-    CONF_TRAVEL_TIMES, RESOURCE, SCAN_INTERVAL)
+    ATTR_DESCRIPTION,
+    ATTR_TIME_UPDATED,
+    CONF_API_KEY,
+    CONF_ID,
+    CONF_NAME,
+    CONF_TRAVEL_TIMES,
+    RESOURCE,
+    SCAN_INTERVAL,
+)
 from homeassistant.setup import setup_component
 
 
@@ -29,11 +36,9 @@ class TestWSDOT(unittest.TestCase):
         """Initialize values for this testcase class."""
         self.hass = get_test_home_assistant()
         self.config = {
-            CONF_API_KEY: 'foo',
+            CONF_API_KEY: "foo",
             SCAN_INTERVAL: timedelta(seconds=120),
-            CONF_TRAVEL_TIMES: [{
-                CONF_ID: 96,
-                CONF_NAME: 'I90 EB'}],
+            CONF_TRAVEL_TIMES: [{CONF_ID: 96, CONF_NAME: "I90 EB"}],
         }
         self.entities = []
 
@@ -43,19 +48,22 @@ class TestWSDOT(unittest.TestCase):
 
     def test_setup_with_config(self):
         """Test the platform setup with configuration."""
-        assert setup_component(self.hass, 'sensor', {'wsdot': self.config})
+        assert setup_component(self.hass, "sensor", {"wsdot": self.config})
 
     @requests_mock.Mocker()
     def test_setup(self, mock_req):
         """Test for operational WSDOT sensor with proper attributes."""
-        uri = re.compile(RESOURCE + '*')
-        mock_req.get(uri, text=load_fixture('wsdot.json'))
+        uri = re.compile(RESOURCE + "*")
+        mock_req.get(uri, text=load_fixture("wsdot.json"))
         wsdot.setup_platform(self.hass, self.config, self.add_entities)
         assert len(self.entities) == 1
         sensor = self.entities[0]
-        assert sensor.name == 'I90 EB'
+        assert sensor.name == "I90 EB"
         assert sensor.state == 11
-        assert sensor.device_state_attributes[ATTR_DESCRIPTION] == \
-            'Downtown Seattle to Downtown Bellevue via I-90'
-        assert sensor.device_state_attributes[ATTR_TIME_UPDATED] == \
-            datetime(2017, 1, 21, 15, 10, tzinfo=timezone(timedelta(hours=-8)))
+        assert (
+            sensor.device_state_attributes[ATTR_DESCRIPTION]
+            == "Downtown Seattle to Downtown Bellevue via I-90"
+        )
+        assert sensor.device_state_attributes[ATTR_TIME_UPDATED] == datetime(
+            2017, 1, 21, 15, 10, tzinfo=timezone(timedelta(hours=-8))
+        )