Commit afd5ede5 authored by fbourgin's avatar fbourgin
Browse files

add a test for q_lvl

parent 38577cd9
1 merge request!4Resolve "evaluation of quantiles prediction"
Pipeline #63737 failed with stage
in 5 minutes and 9 seconds
Showing with 23 additions and 0 deletions
+23 -0
0.00643087,0.0514469
......@@ -90,6 +90,13 @@ class TestMetrics(unittest.TestCase):
) for metric in ('CR', 'AW', 'AWN', 'WS')
}
expected_itv_qlvl = {
metric: (
numpy.genfromtxt(f"./expected/evalp/{metric}_QLVL.csv", delimiter=',')
[numpy.newaxis, numpy.newaxis, numpy.newaxis, numpy.newaxis, ...]
) for metric in ('CR')
}
expected_mvr = {
metric: (
numpy.genfromtxt(f"./expected/evalp/{metric}.csv", delimiter=',')
......@@ -153,6 +160,21 @@ class TestMetrics(unittest.TestCase):
self.expected_itv[metric]
)
def test_intervals_qlvl_metrics(self):
lvl = numpy.array([50., 80.])
qlvl = numpy.array(10., 25., 75., 90])
for metric in expected_itv_qlvl.keys():
numpy.set_printoptions(precision=13)
m = evalhyd.evalp(_obs, _prd, [metric], c_lvl=lvl, q_lvl=qlvl)[0][0, 0, 0]
numpy.savetxt(f"./expected/evalp/{metric}_QLVL.csv", m, delimiter=',', fmt="%.13f")
with self.subTest(metric=metric):
numpy.testing.assert_almost_equal(
evalhyd.evalp(_obs, _prd, [metric], c_lvl=lvl, q_lvl=qlvl)[0],
self.expected_itv_qlvl[metric]
)
def test_multivariate_metrics(self):
n_sit = 5
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment