diff --git a/tests/test_sonoff.py b/tests/test_sonoff.py index 17b340b3f7..d4b39406d6 100644 --- a/tests/test_sonoff.py +++ b/tests/test_sonoff.py @@ -1,9 +1,5 @@ """Tests for Sonoff quirks.""" -from unittest import mock - -import pytest - import zhaquirks import zhaquirks.sonoff.swv diff --git a/zhaquirks/sonoff/swv.py b/zhaquirks/sonoff/swv.py index f9637c7e07..304e6521da 100644 --- a/zhaquirks/sonoff/swv.py +++ b/zhaquirks/sonoff/swv.py @@ -5,11 +5,11 @@ import zigpy.types as t from zigpy.zcl.clusters.general import ( Basic, - PowerConfiguration, Identify, OnOff, - PollControl, Ota, + PollControl, + PowerConfiguration, Time, ) from zigpy.zcl.clusters.homeautomation import Diagnostic