Convert fan tests to pytest style (#41631)

This commit is contained in:
Sharif Nassar 2020-10-14 02:07:26 -07:00 committed by GitHub
parent afc602595b
commit 9ee97cb213
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,7 +1,5 @@
"""Tests for fan platforms.""" """Tests for fan platforms."""
import unittest
import pytest import pytest
from homeassistant.components.fan import FanEntity from homeassistant.components.fan import FanEntity
@ -12,32 +10,20 @@ class BaseFan(FanEntity):
def __init__(self): def __init__(self):
"""Initialize the fan.""" """Initialize the fan."""
pass
class TestFanEntity(unittest.TestCase): def test_fanentity():
"""Test coverage for base fan entity class.""" """Test fan entity methods."""
fan = BaseFan()
def setUp(self): assert fan.state == "off"
"""Set up test data.""" assert len(fan.speed_list) == 0
self.fan = BaseFan() assert fan.supported_features == 0
self.addCleanup(self.tear_down_cleanup) assert fan.capability_attributes == {}
# Test set_speed not required
def tear_down_cleanup(self): fan.oscillate(True)
"""Tear down unit test data.""" with pytest.raises(NotImplementedError):
self.fan = None fan.set_speed("slow")
with pytest.raises(NotImplementedError):
def test_fanentity(self): fan.turn_on()
"""Test fan entity methods.""" with pytest.raises(NotImplementedError):
assert self.fan.state == "off" fan.turn_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()