From cb32f77130094a1035ed7e202c28469b768c0be2 Mon Sep 17 00:00:00 2001 From: Jules Dejaeghere Date: Wed, 3 Apr 2024 11:27:06 +0200 Subject: [PATCH] Add test for SVG parsing --- tests/fixtures/pollen_two.svg | 2 ++ tests/test_pollen.py | 6 ++++++ 2 files changed, 8 insertions(+) create mode 100644 tests/fixtures/pollen_two.svg 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']