| import unittest |
| from benchmark_intensity_v12_v11 import aggregate, common_errors, initial_available |
|
|
|
|
| class IntensityProtocolTests(unittest.TestCase): |
| def test_initial_missing_pressure_is_not_filled(self): |
| self.assertFalse(initial_available([40, 0])) |
| self.assertFalse(initial_available([float('nan'), 990])) |
| self.assertTrue(initial_available([0, 1000])) |
|
|
| def test_common_masks_retain_zero_wind(self): |
| result = common_errors([0, 2, 3], [1, float('nan'), 4], [0, 1, None], low=0, high=249) |
| self.assertEqual(result, {'1.1': [0., None, None], '1.2': [1., None, None]}) |
|
|
| def test_equal_storm_not_equal_issue(self): |
| rows = [] |
| for sid, old, new in [('a', 10, 4), ('a', 10, 4), ('b', 20, 12)]: |
| rows.append({'storm_id': sid, 'errors': {'p': {'1.1': [old]*20, '1.2': [new]*20}}}) |
| report = aggregate(rows, 'p') |
| self.assertEqual(report['models']['1.1']['equal_storm_mae'], 15) |
| self.assertEqual(report['models']['1.2']['equal_storm_mae'], 8) |
| self.assertEqual(report['valid_daily_issues'], 3) |
|
|
| def test_missing_leads_remain_unavailable(self): |
| report = aggregate([{'storm_id': 'a', 'errors': {'p': {'1.1': [1]+[None]*19, '1.2': [2]+[None]*19}}}], 'p') |
| self.assertEqual(report['models']['1.1']['equal_storm_mae_by_lead'], [1]+[None]*19) |
| self.assertEqual(report['paired_issue_leads_by_lead'], [1]+[0]*19) |
|
|
|
|
| if __name__ == '__main__': |
| unittest.main() |
|
|
