diff --git a/homeassistant/components/climate/nuheat.py b/homeassistant/components/climate/nuheat.py index 226c92a6e65..a4d1cad68a4 100644 --- a/homeassistant/components/climate/nuheat.py +++ b/homeassistant/components/climate/nuheat.py @@ -95,7 +95,6 @@ def setup_platform(hass, config, add_devices, discovery_info=None): schema=RESUME_PROGRAM_SCHEMA) - class NuHeatThermostat(ClimateDevice): """Representation of a NuHeat Thermostat.""" diff --git a/tests/components/climate/test_nuheat.py b/tests/components/climate/test_nuheat.py index 3e30e7caaf7..aedb925277e 100644 --- a/tests/components/climate/test_nuheat.py +++ b/tests/components/climate/test_nuheat.py @@ -141,15 +141,18 @@ class TestNuHeat(unittest.TestCase): def test_set_hold_mode(self): """Test set hold mode.""" self.thermostat.set_hold_mode("temperature") - self.assertEqual(self.thermostat._thermostat.schedule_mode, SCHEDULE_HOLD) + self.assertEqual( + self.thermostat._thermostat.schedule_mode, SCHEDULE_HOLD) self.assertTrue(self.thermostat._force_update) self.thermostat.set_hold_mode("temporary_temperature") - self.assertEqual(self.thermostat._thermostat.schedule_mode, SCHEDULE_TEMPORARY_HOLD) + self.assertEqual( + self.thermostat._thermostat.schedule_mode, SCHEDULE_TEMPORARY_HOLD) self.assertTrue(self.thermostat._force_update) self.thermostat.set_hold_mode("auto") - self.assertEqual(self.thermostat._thermostat.schedule_mode, SCHEDULE_RUN) + self.assertEqual( + self.thermostat._thermostat.schedule_mode, SCHEDULE_RUN) self.assertTrue(self.thermostat._force_update) def test_set_temperature(self):