one more sanity check

NREL · Mar 28, 2024 · 226b67f · 226b67f
1 parent 03ef3b2
commit 226b67f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/bias/test_bias_correction.py b/tests/bias/test_bias_correction.py
@@ -545,10 +545,13 @@ def test_match_zero_rate():
  assert skill['bias_f1_zero_rate'] != skill['base_f1_zero_rate']
  assert (bias_data == 0).mean() != (base_data == 0).mean()
 
- bias_data = SkillAssessment._match_zero_rate(bias_data, base_data)
- skill = SkillAssessment._run_skill_eval(bias_data, base_data, 'f1', 'f1')
+ skill = SkillAssessment._run_skill_eval(bias_data, base_data, 'f1', 'f1',
+  match_zero_rate=True)
  assert (bias_data == 0).mean() == (base_data == 0).mean()
  assert skill['bias_f1_zero_rate'] == skill['base_f1_zero_rate']
+ for p in (1, 5, 25, 50, 75, 95, 99):
+ assert np.allclose(skill[f'base_f1_percentile_{p}'],
+ np.percentile(base_data, p))
 
  with tempfile.TemporaryDirectory() as td:
  fp_nsrdb_temp = os.path.join(td, os.path.basename(FP_NSRDB))