first commit

api/tests/test_core.py

@@ -0,0 +1,56 @@
+from math import isclose
+
+import pytest
+
+from app.core.antenna import AntennaPattern, gain
+from app.core.diffraction import knife_edge_loss
+from app.core.fresnel import earth_bulge, fresnel_radius, wavelength
+from app.core.propagation import fspl
+from app.core.vegetation import P833Coefficients, p833_attenuation
+
+
+def test_fspl_reference_value() -> None:
+    assert isclose(fspl(433, 12.45), 107.08, abs_tol=0.05)
+
+
+def test_fresnel_radius_reference_value() -> None:
+    lmbda = wavelength(433_000_000)
+    assert isclose(fresnel_radius(lmbda, 6_225, 6_225), 46.45, abs_tol=0.05)
+
+
+def test_earth_bulge_reference_value() -> None:
+    assert isclose(earth_bulge(6_225, 6_225, 1.333), 2.28, abs_tol=0.02)
+
+
+@pytest.mark.parametrize(
+    ("v", "expected"),
+    [
+        (-0.79, 0.0),
+        (0.0, 6.03),
+        (1.0, 13.93),
+    ],
+)
+def test_knife_edge_loss_reference_values(v: float, expected: float) -> None:
+    assert isclose(knife_edge_loss(v), expected, abs_tol=0.05)
+
+
+def test_p833_formula_with_supplied_coefficients() -> None:
+    attenuation = p833_attenuation(
+        depth_m=100,
+        freq_hz=433_000_000,
+        forest_type="coniferous",
+        coefficients=P833Coefficients(gamma_db_per_m=0.2, max_attenuation_db=30),
+    )
+    assert isclose(attenuation, 14.60, abs_tol=0.05)
+
+
+def test_sector_antenna_gain_peaks_at_boresight() -> None:
+    pattern = AntennaPattern(
+        pattern="sector",
+        azimuth_deg=90,
+        gain_dbi=8,
+        beamwidth_h=65,
+        beamwidth_v=15,
+    )
+    assert gain(pattern, 90, 0) == 8
+    assert gain(pattern, 270, 0) < 8