diff --git a/tests/components/fan/test_init.py b/tests/components/fan/test_init.py index 134a5976baa..c481e9f0937 100644 --- a/tests/components/fan/test_init.py +++ b/tests/components/fan/test_init.py @@ -1,7 +1,5 @@ """Tests for fan platforms.""" -import unittest - import pytest from homeassistant.components.fan import FanEntity @@ -12,32 +10,20 @@ class BaseFan(FanEntity): def __init__(self): """Initialize the fan.""" - pass -class TestFanEntity(unittest.TestCase): - """Test coverage for base fan entity class.""" - - def setUp(self): - """Set up test data.""" - self.fan = BaseFan() - self.addCleanup(self.tear_down_cleanup) - - def tear_down_cleanup(self): - """Tear down unit test data.""" - self.fan = None - - def test_fanentity(self): - """Test fan entity methods.""" - assert self.fan.state == "off" - assert len(self.fan.speed_list) == 0 - assert self.fan.supported_features == 0 - assert self.fan.capability_attributes == {} - # Test set_speed not required - self.fan.oscillate(True) - with pytest.raises(NotImplementedError): - self.fan.set_speed("slow") - with pytest.raises(NotImplementedError): - self.fan.turn_on() - with pytest.raises(NotImplementedError): - self.fan.turn_off() +def test_fanentity(): + """Test fan entity methods.""" + fan = BaseFan() + assert fan.state == "off" + assert len(fan.speed_list) == 0 + assert fan.supported_features == 0 + assert fan.capability_attributes == {} + # Test set_speed not required + fan.oscillate(True) + with pytest.raises(NotImplementedError): + fan.set_speed("slow") + with pytest.raises(NotImplementedError): + fan.turn_on() + with pytest.raises(NotImplementedError): + fan.turn_off()