Add tests

This commit is contained in:
Jules 2024-04-01 18:13:16 +02:00
parent c8b2954ef3
commit 43a207522a
Signed by: jdejaegh
GPG key ID: 99D6D184CA66933A

View file

@ -1,7 +1,18 @@
from custom_components.irm_kmi.pollen import PollenParser from custom_components.irm_kmi.pollen import PollenParser
def test_svg_pollen_parsing(): def test_svg_pollen_parsing():
# TODO make it an actual test
with open("tests/fixtures/pollen.svg", "r") as file: with open("tests/fixtures/pollen.svg", "r") as file:
svg_data = file.read() svg_data = file.read()
PollenParser(svg_data).get_pollen_data() data = PollenParser(svg_data).get_pollen_data()
assert data == {'birch': 'purple', 'oak': 'active', 'hazel': 'none', 'mugwort': 'none', 'alder': 'green',
'grasses': 'none', 'ash': 'active'}
def test_pollen_options():
assert PollenParser.get_option_values() == ['active', 'green', 'yellow', 'orange', 'red', 'purple', 'none']
def test_pollen_default_values():
assert PollenParser.get_default_data() == {'birch': 'none', 'oak': 'none', 'hazel': 'none', 'mugwort': 'none',
'alder': 'none', 'grasses': 'none', 'ash': 'none'}