diff --git a/tests/test_perf.py b/tests/test_perf.py index 4153e6d..48b46d6 100644 --- a/tests/test_perf.py +++ b/tests/test_perf.py @@ -27,10 +27,10 @@ def inner(): benchmark(inner) -def test_calc_mae_1e9(benchmark): +def test_calc_mae_1e7(benchmark): def inner(): - obs = np.random.rand(int(1e9)) - fct = np.random.rand(int(1e9)) + obs = np.random.rand(int(1e7)) + fct = np.random.rand(int(1e7)) calc_mae(obs, fct) benchmark(inner) @@ -54,10 +54,10 @@ def inner(): benchmark(inner) -def test_calc_mbe_1e9(benchmark): +def test_calc_mbe_1e7(benchmark): def inner(): - obs = np.random.rand(int(1e9)) - fct = np.random.rand(int(1e9)) + obs = np.random.rand(int(1e7)) + fct = np.random.rand(int(1e7)) calc_mbe(obs, fct) benchmark(inner) @@ -81,10 +81,10 @@ def inner(): benchmark(inner) -def test_calc_threshold_hit_ratio_1e9(benchmark): +def test_calc_threshold_hit_ratio_1e7(benchmark): def inner(): - obs = np.random.rand(int(1e9)) - fct = np.random.rand(int(1e9)) + obs = np.random.rand(int(1e7)) + fct = np.random.rand(int(1e7)) calc_threshold_hit_ratio(obs, fct, 0.5) benchmark(inner) @@ -108,10 +108,10 @@ def inner(): benchmark(inner) -def test_calc_threshold_false_alarm_ratio_1e9(benchmark): +def test_calc_threshold_false_alarm_ratio_1e7(benchmark): def inner(): - obs = np.random.rand(int(1e9)) - fct = np.random.rand(int(1e9)) + obs = np.random.rand(int(1e7)) + fct = np.random.rand(int(1e7)) calc_threshold_false_alarm_ratio(obs, fct, 0.5) benchmark(inner) @@ -135,10 +135,10 @@ def inner(): benchmark(inner) -def test_calc_threshold_ts_1e9(benchmark): +def test_calc_threshold_ts_1e7(benchmark): def inner(): - obs = np.random.rand(int(1e9)) - fct = np.random.rand(int(1e9)) + obs = np.random.rand(int(1e7)) + fct = np.random.rand(int(1e7)) calc_threshold_ts(obs, fct, 0.5) benchmark(inner)