diff --git a/tests/fixtures/pollen_two.svg b/tests/fixtures/pollen_two.svg new file mode 100644 index 0000000..4bfe1b6 --- /dev/null +++ b/tests/fixtures/pollen_two.svg @@ -0,0 +1,2 @@ + + Active pollen Alder activeAsh activeOak active Birch \ No newline at end of file diff --git a/tests/test_pollen.py b/tests/test_pollen.py index e715419..b53f2b9 100644 --- a/tests/test_pollen.py +++ b/tests/test_pollen.py @@ -15,6 +15,12 @@ def test_svg_pollen_parsing(): assert data == {'birch': 'purple', 'oak': 'active', 'hazel': 'none', 'mugwort': 'none', 'alder': 'green', 'grasses': 'none', 'ash': 'active'} + with open("tests/fixtures/pollen_two.svg", "r") as file: + svg_data = file.read() + data = PollenParser(svg_data).get_pollen_data() + assert data == {'birch': 'purple', 'oak': 'active', 'hazel': 'none', 'mugwort': 'none', 'alder': 'active', + 'grasses': 'none', 'ash': 'active'} + def test_pollen_options(): assert PollenParser.get_option_values() == ['active', 'green', 'yellow', 'orange', 'red', 'purple', 'none']