run black

Author: basnijholt

custom_components/sonoff/core/devices.py

@@ -17,7 +17,6 @@
 from homeassistant.components.sensor import SensorEntity
 from homeassistant.components.switch import SwitchEntity
 
-from .ewelink import XDevice
 from ..binary_sensor import XBinarySensor, XWiFiDoor, XZigbeeMotion
 from ..climate import XClimateNS, XClimateTH, XThermostat
 from ..core.entity import XEntity
@@ -39,6 +38,7 @@
 from ..remote import XRemote
 from ..sensor import (
     XEnergySensor,
+    XEnergySensorDualR3,
     XHumidityTH,
     XOutdoorTempNS,
     XRemoteButton,
@@ -47,9 +47,9 @@
     XTemperatureTH,
     XUnknown,
     XWiFiDoorBattery,
-    XEnergySensorDualR3,
 )
 from ..switch import XSwitch, XSwitches, XSwitchTH, XToggle, XZigbeeSwitches
+from .ewelink import XDevice
 
 # supported custom device_class
 DEVICE_CLASS = {

tests/test_entity.py

@@ -450,13 +450,9 @@ def test_dual_r3():
                 "getKwh_01": 2,
                 "calibration": 1,
             },
-    }, {
-        "devices": {
-            DEVICEID: {
-                "reporting": {"energy_1": [3600, 3]}
-            }
-        }
-    })
+        },
+        {"devices": {DEVICEID: {"reporting": {"energy_1": [3600, 3]}}}},
+    )
 
     volt: XSensor = next(e for e in entities if e.uid == "voltage_1")
     assert volt.state == 247.62
@@ -469,15 +465,15 @@ def test_dual_r3():
 
     # Get history if we use reporting
     energy_1: XEnergySensor_DualR3 = next(e for e in entities if e.uid == "energy_1")
-    energy_1.internal_update({'kwhHistories_00': '0034007412340000'})
+    energy_1.internal_update({"kwhHistories_00": "0034007412340000"})
     assert energy_1.state == 0.34
     assert energy_1.extra_state_attributes == {"history": [0.34, 0.74, 12.34]}
 
     # Skip history if we don't use reporting
     energy_2: XEnergySensor_DualR3 = next(e for e in entities if e.uid == "energy_2")
-    energy_2.internal_update({'kwhHistories_01': '0201000000000000'})
+    energy_2.internal_update({"kwhHistories_01": "0201000000000000"})
     assert energy_2.state == 2.01
-    assert energy_2.extra_state_attributes == None
+    assert energy_2.extra_state_attributes is None
 
 
 def test_diffuser():