diff --git a/tests/components/mfi/test_sensor.py b/tests/components/mfi/test_sensor.py
index 6103c43d3a4..1f5cc5fd04f 100644
--- a/tests/components/mfi/test_sensor.py
+++ b/tests/components/mfi/test_sensor.py
@@ -7,7 +7,7 @@ import requests
 
 import homeassistant.components.mfi.sensor as mfi
 import homeassistant.components.sensor as sensor_component
-from homeassistant.const import TEMP_CELSIUS
+from homeassistant.const import DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS
 from homeassistant.setup import async_setup_component
 
 PLATFORM = mfi
@@ -133,30 +133,35 @@ async def test_uom_temp(port, sensor):
     """Test the UOM temperature."""
     port.tag = "temperature"
     assert sensor.unit_of_measurement == TEMP_CELSIUS
+    assert sensor.device_class == DEVICE_CLASS_TEMPERATURE
 
 
 async def test_uom_power(port, sensor):
     """Test the UOEM power."""
     port.tag = "active_pwr"
     assert sensor.unit_of_measurement == "Watts"
+    assert sensor.device_class is None
 
 
 async def test_uom_digital(port, sensor):
     """Test the UOM digital input."""
     port.model = "Input Digital"
     assert sensor.unit_of_measurement == "State"
+    assert sensor.device_class is None
 
 
 async def test_uom_unknown(port, sensor):
     """Test the UOM."""
     port.tag = "balloons"
     assert sensor.unit_of_measurement == "balloons"
+    assert sensor.device_class is None
 
 
 async def test_uom_uninitialized(port, sensor):
     """Test that the UOM defaults if not initialized."""
     type(port).tag = mock.PropertyMock(side_effect=ValueError)
     assert sensor.unit_of_measurement == "State"
+    assert sensor.device_class is None
 
 
 async def test_state_digital(port, sensor):
diff --git a/tests/components/zwave/test_sensor.py b/tests/components/zwave/test_sensor.py
index b2cd895df37..ae0fa44ed8c 100644
--- a/tests/components/zwave/test_sensor.py
+++ b/tests/components/zwave/test_sensor.py
@@ -84,6 +84,7 @@ def test_multilevelsensor_value_changed_temp_fahrenheit(mock_openzwave):
     device = sensor.get_device(node=node, values=values, node_config={})
     assert device.state == 191.0
     assert device.unit_of_measurement == homeassistant.const.TEMP_FAHRENHEIT
+    assert device.device_class == homeassistant.const.DEVICE_CLASS_TEMPERATURE
     value.data = 197.95555
     value_changed(value)
     assert device.state == 198.0
@@ -103,6 +104,7 @@ def test_multilevelsensor_value_changed_temp_celsius(mock_openzwave):
     device = sensor.get_device(node=node, values=values, node_config={})
     assert device.state == 38.9
     assert device.unit_of_measurement == homeassistant.const.TEMP_CELSIUS
+    assert device.device_class == homeassistant.const.DEVICE_CLASS_TEMPERATURE
     value.data = 37.95555
     value_changed(value)
     assert device.state == 38.0
@@ -124,6 +126,7 @@ def test_multilevelsensor_value_changed_other_units(mock_openzwave):
     device = sensor.get_device(node=node, values=values, node_config={})
     assert device.state == 190.96
     assert device.unit_of_measurement == homeassistant.const.ENERGY_KILO_WATT_HOUR
+    assert device.device_class is None
     value.data = 197.95555
     value_changed(value)
     assert device.state == 197.96
@@ -143,6 +146,7 @@ def test_multilevelsensor_value_changed_integer(mock_openzwave):
     device = sensor.get_device(node=node, values=values, node_config={})
     assert device.state == 5
     assert device.unit_of_measurement == "counts"
+    assert device.device_class is None
     value.data = 6
     value_changed(value)
     assert device.state == 6
@@ -159,6 +163,7 @@ def test_alarm_sensor_value_changed(mock_openzwave):
     device = sensor.get_device(node=node, values=values, node_config={})
     assert device.state == 12.34
     assert device.unit_of_measurement == homeassistant.const.PERCENTAGE
+    assert device.device_class is None
     value.data = 45.67
     value_changed(value)
     assert device.state == 45.67