Use new enums in zwave tests (#62711)

* Use new enums in zwave tests

* Code review: == to is and BinarySensor to Sensor
This commit is contained in:
Dave T 2021-12-23 22:51:33 +00:00 committed by GitHub
parent 759481688b
commit fa6d6d914b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,4 +1,5 @@
"""Test Z-Wave sensor."""
from homeassistant.components.sensor import SensorDeviceClass
from homeassistant.components.zwave import const, sensor
import homeassistant.const
@ -67,7 +68,7 @@ def test_get_device_detects_battery_sensor(mock_openzwave):
device = sensor.get_device(node=node, values=values, node_config={})
assert isinstance(device, sensor.ZWaveBatterySensor)
assert device.device_class == homeassistant.const.DEVICE_CLASS_BATTERY
assert device.device_class is SensorDeviceClass.BATTERY
def test_multilevelsensor_value_changed_temp_fahrenheit(hass, mock_openzwave):
@ -87,7 +88,7 @@ def test_multilevelsensor_value_changed_temp_fahrenheit(hass, mock_openzwave):
device.hass = hass
assert device.state == 191.0
assert device.unit_of_measurement == homeassistant.const.TEMP_FAHRENHEIT
assert device.device_class == homeassistant.const.DEVICE_CLASS_TEMPERATURE
assert device.device_class is SensorDeviceClass.TEMPERATURE
value.data = 197.95555
value_changed(value)
assert device.state == 198.0
@ -109,7 +110,7 @@ def test_multilevelsensor_value_changed_temp_celsius(hass, mock_openzwave):
device.hass = hass
assert device.state == 38.9
assert device.unit_of_measurement == homeassistant.const.TEMP_CELSIUS
assert device.device_class == homeassistant.const.DEVICE_CLASS_TEMPERATURE
assert device.device_class is SensorDeviceClass.TEMPERATURE
value.data = 37.95555
value_changed(value)
assert device.state == 38.0