From eb8fb5d6d47b7fb570603693f6bbd368a4d8e639 Mon Sep 17 00:00:00 2001
From: Jules Dejaeghere <dej.j@live.be>
Date: Thu, 20 Jun 2024 22:02:02 +0200
Subject: [PATCH] Add tests

---
 tests/test_belaqi.py | 87 ++++++++++++++++++++++++++++++++++++++++++++
 tests/test_utils.py  |  2 +-
 2 files changed, 88 insertions(+), 1 deletion(-)

diff --git a/tests/test_belaqi.py b/tests/test_belaqi.py
index 0997d70..f4ffee4 100644
--- a/tests/test_belaqi.py
+++ b/tests/test_belaqi.py
@@ -1,3 +1,5 @@
+from random import randint, seed
+import pytest
 from src.open_irceline.belaqi import belaqi_index
 from src.open_irceline.data import BelAqiIndex
 
@@ -52,3 +54,88 @@ def test_belaqi_index():
     assert belaqi_index(110, 75, 330, 310) == BelAqiIndex.HORRIBLE
     assert belaqi_index(101, 71, 321, 301) == BelAqiIndex.HORRIBLE
     assert belaqi_index(150, 100, 400, 400) == BelAqiIndex.HORRIBLE
+
+
+def test_belaqi_single_component():
+    # Tests with only PM10 varying
+    assert belaqi_index(5, 0, 0, 0) == BelAqiIndex.EXCELLENT
+    assert belaqi_index(15, 0, 0, 0) == BelAqiIndex.VERY_GOOD
+    assert belaqi_index(25, 0, 0, 0) == BelAqiIndex.GOOD
+    assert belaqi_index(35, 0, 0, 0) == BelAqiIndex.FAIRLY_GOOD
+    assert belaqi_index(45, 0, 0, 0) == BelAqiIndex.MODERATE
+    assert belaqi_index(55, 0, 0, 0) == BelAqiIndex.POOR
+    assert belaqi_index(65, 0, 0, 0) == BelAqiIndex.VERY_POOR
+    assert belaqi_index(75, 0, 0, 0) == BelAqiIndex.BAD
+    assert belaqi_index(85, 0, 0, 0) == BelAqiIndex.VERY_BAD
+    assert belaqi_index(110, 0, 0, 0) == BelAqiIndex.HORRIBLE
+
+    # Tests with only PM2.5 varying
+    assert belaqi_index(0, 2, 0, 0) == BelAqiIndex.EXCELLENT
+    assert belaqi_index(0, 8, 0, 0) == BelAqiIndex.VERY_GOOD
+    assert belaqi_index(0, 12, 0, 0) == BelAqiIndex.GOOD
+    assert belaqi_index(0, 20, 0, 0) == BelAqiIndex.FAIRLY_GOOD
+    assert belaqi_index(0, 30, 0, 0) == BelAqiIndex.MODERATE
+    assert belaqi_index(0, 38, 0, 0) == BelAqiIndex.POOR
+    assert belaqi_index(0, 45, 0, 0) == BelAqiIndex.VERY_POOR
+    assert belaqi_index(0, 55, 0, 0) == BelAqiIndex.BAD
+    assert belaqi_index(0, 65, 0, 0) == BelAqiIndex.VERY_BAD
+    assert belaqi_index(0, 75, 0, 0) == BelAqiIndex.HORRIBLE
+
+    # Tests with only O3 varying
+    assert belaqi_index(0, 0, 10, 0) == BelAqiIndex.EXCELLENT
+    assert belaqi_index(0, 0, 40, 0) == BelAqiIndex.VERY_GOOD
+    assert belaqi_index(0, 0, 60, 0) == BelAqiIndex.GOOD
+    assert belaqi_index(0, 0, 100, 0) == BelAqiIndex.FAIRLY_GOOD
+    assert belaqi_index(0, 0, 140, 0) == BelAqiIndex.MODERATE
+    assert belaqi_index(0, 0, 170, 0) == BelAqiIndex.POOR
+    assert belaqi_index(0, 0, 200, 0) == BelAqiIndex.VERY_POOR
+    assert belaqi_index(0, 0, 260, 0) == BelAqiIndex.BAD
+    assert belaqi_index(0, 0, 300, 0) == BelAqiIndex.VERY_BAD
+    assert belaqi_index(0, 0, 330, 0) == BelAqiIndex.HORRIBLE
+
+    # Tests with only NO2 varying
+    assert belaqi_index(0, 0, 0, 10) == BelAqiIndex.EXCELLENT
+    assert belaqi_index(0, 0, 0, 35) == BelAqiIndex.VERY_GOOD
+    assert belaqi_index(0, 0, 0, 60) == BelAqiIndex.GOOD
+    assert belaqi_index(0, 0, 0, 90) == BelAqiIndex.FAIRLY_GOOD
+    assert belaqi_index(0, 0, 0, 130) == BelAqiIndex.MODERATE
+    assert belaqi_index(0, 0, 0, 160) == BelAqiIndex.POOR
+    assert belaqi_index(0, 0, 0, 190) == BelAqiIndex.VERY_POOR
+    assert belaqi_index(0, 0, 0, 220) == BelAqiIndex.BAD
+    assert belaqi_index(0, 0, 0, 270) == BelAqiIndex.VERY_BAD
+    assert belaqi_index(0, 0, 0, 310) == BelAqiIndex.HORRIBLE
+
+
+def test_belaqi_random():
+    seed(42)
+    # Generate random test values and their expected indices
+    test_cases = [
+        (randint(0, 10), randint(0, 5), randint(0, 25), randint(0, 20), BelAqiIndex.EXCELLENT),
+        (randint(11, 20), randint(6, 10), randint(26, 50), randint(21, 50), BelAqiIndex.VERY_GOOD),
+        (randint(21, 30), randint(11, 15), randint(51, 70), randint(51, 70), BelAqiIndex.GOOD),
+        (randint(31, 40), randint(16, 25), randint(71, 120), randint(71, 120), BelAqiIndex.FAIRLY_GOOD),
+        (randint(41, 50), randint(26, 35), randint(121, 160), randint(121, 150), BelAqiIndex.MODERATE),
+        (randint(51, 60), randint(36, 40), randint(161, 180), randint(151, 180), BelAqiIndex.POOR),
+        (randint(61, 70), randint(41, 50), randint(181, 240), randint(181, 200), BelAqiIndex.VERY_POOR),
+        (randint(71, 80), randint(51, 60), randint(241, 280), randint(201, 250), BelAqiIndex.BAD),
+        (randint(81, 100), randint(61, 70), randint(281, 320), randint(251, 300), BelAqiIndex.VERY_BAD),
+        (randint(101, 150), randint(71, 100), randint(321, 400), randint(301, 400), BelAqiIndex.HORRIBLE)
+    ]
+
+    # Test each case
+    for pm10, pm25, o3, no2, expected in test_cases:
+        assert belaqi_index(pm10, pm25, o3, no2) == expected
+
+
+def test_belaqi_value_error():
+    with pytest.raises(ValueError):
+        belaqi_index(-1, 0, 12, 8)
+
+    with pytest.raises(ValueError):
+        belaqi_index(1, -20, 12, 8)
+
+    with pytest.raises(ValueError):
+        belaqi_index(1, 0, -12, 8)
+
+    with pytest.raises(ValueError):
+        belaqi_index(1, 0, 12, -8888)
diff --git a/tests/test_utils.py b/tests/test_utils.py
index e2a9b1a..10bba8b 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -2,7 +2,7 @@ from src.open_irceline.utils import SizedDict, round_coordinates, epsg_transform
 import pytest
 
 
-def test_size_dict():
+def test_sized_dict():
     s_dict = SizedDict(5)
     assert len(s_dict) == 0