Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
40 changes: 29 additions & 11 deletions tests/test_weather.py
Original file line number Diff line number Diff line change
@@ -1,24 +1,42 @@
import math

import pytest

from WeatherRoutingTool.weather import WeatherCond


SQRT2_OVER_2 = math.sqrt(2) / 2


@pytest.mark.parametrize("u,v,theta_res", [(-1, -1, 45), (1, -1, 315), (1, 1, 225), (-1, 1, 135)])
def test_theta_from_uv(u, v, theta_res):
theta_test = WeatherCond.get_theta_from_uv(u, v)

assert theta_test == pytest.approx(theta_res)


@pytest.mark.parametrize("theta,windspeed,u_res", [(45, 1, -1), (315, 1, 1), (225, 1, 1), (135, 1, -1)])
def get_u(theta, windspeed):
@pytest.mark.parametrize(
"theta,windspeed,u_res",
[
(45, 1, -SQRT2_OVER_2),
(315, 1, SQRT2_OVER_2),
(225, 1, SQRT2_OVER_2),
(135, 1, -SQRT2_OVER_2),
],
)
def test_get_u(theta, windspeed, u_res):
u_test = WeatherCond.get_u(theta, windspeed)

assert u_test == pytest.approx(u_test)


@pytest.mark.parametrize("theta,windspeed,u_res", [(45, 1, -1), (315, 1, -1), (225, 1, 1), (135, 1, 1)])
def get_v(theta, windspeed):
assert u_test == pytest.approx(u_res)


@pytest.mark.parametrize(
"theta,windspeed,v_res",
[
(45, 1, -SQRT2_OVER_2),
(315, 1, -SQRT2_OVER_2),
(225, 1, SQRT2_OVER_2),
(135, 1, SQRT2_OVER_2),
],
)
def test_get_v(theta, windspeed, v_res):
v_test = WeatherCond.get_v(theta, windspeed)

assert v_test == pytest.approx(v_test)
assert v_test == pytest.approx(v_res)