소스 검색

tests: add missing imports

Jason Rumney 6 달 전
부모
커밋
73685ea40a
2개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      tests/devices/test_duux_blizzard.py
  2. 1 0
      tests/devices/test_smartplugv2_energy.py

+ 1 - 0
tests/devices/test_duux_blizzard.py

@@ -1,6 +1,7 @@
 from homeassistant.components.climate.const import HVACMode
 
 from ..const import DUUX_BLIZZARD_PAYLOAD
+from ..helpers import assert_device_properties_set
 from ..mixins.climate import TargetTemperatureTests
 from .base_device_tests import TuyaDeviceTestCase
 

+ 1 - 0
tests/devices/test_smartplugv2_energy.py

@@ -13,6 +13,7 @@ from homeassistant.const import (
 )
 
 from ..const import SMARTSWITCH_ENERGY_PAYLOAD
+from ..helpers import assert_device_properties_set
 from ..mixins.binary_sensor import BasicBinarySensorTests
 from ..mixins.number import BasicNumberTests
 from ..mixins.select import BasicSelectTests