def test_nan_submission(self):
errors = validate_submission("2020-08-01", "2020-08-04", IP_FILE_FEW_COUNTRIES, INVALID_RANGE_SUBMISSION)
self.assertIsNotNone(errors)
self.assertTrue("Column C1" in errors[0], f"Expected Column C1 in errors, but got {errors}")
self.assertTrue("NaN" in errors[0], f"Expected 'NaN' in errors, but got {errors}")
self.assertTrue("Column C2" in errors[1], f"Expected Column C2 in errors, but got {errors}")
self.assertTrue("negative" in errors[1], f"Expected 'negative' in errors, but got {errors}")
self.assertTrue("Column C3" in errors[2], f"Expected Column C3 in errors, but got {errors}")
self.assertTrue("higher than max" in errors[2], f"Expected 'higher than max' in errors, but got {errors}")
def test_dates(self):
errors = validate_submission("2020-08-01", "2020-08-04", IP_FILE_FEW_COUNTRIES, BAD_DATES_SUBMISSION)
self.assertIsNotNone(errors)
self.assertEqual(10, len(errors), "Not the expected number of errors")
expected_errors = ['Afghanistan: Expected prediction for date 2020-08-01 but got 2020-07-01',
'Afghanistan: Expected prediction for date 2020-08-02 but got 2020-07-02',
'Afghanistan: Expected prediction for date 2020-08-03 but got 2020-07-03',
'Afghanistan: Expected prediction for date 2020-08-04 but got 2020-07-04',
'Albania: Expected prediction for date 2020-08-03 but got None',
'Albania: Expected prediction for date 2020-08-04 but got None',
'Angola: Expected prediction for date None but got 2020-08-05',
'Aruba: Expected prediction for date 2020-08-02 but got 2020-08-03',
'Aruba: Expected prediction for date 2020-08-03 but got 2020-08-04',
'Aruba: Expected prediction for date 2020-08-04 but got None']
self.assertEqual(expected_errors, errors)
