validation_test_wrapper

validation_test_wrapper.py

def test_data(df, tests):
    """Run provided data tests on provided data."""
    results = []
    for test_func, failure_message in tests:
        results.append(test_func(df.copy()))
        if results[-1]:
            logger.info(f'Data test {test_func.__name__} passed.')
        else:
            logger.error(f'Data test {test_func.__name__} failed. {failure_message}')
    logger.info(f'{sum(results)}/{len(results)} passed.')
    return sum(results) == len(results)
  
    
nyt_cases_counties = [
    (cases_vs_deaths, "Death counts cannot exceed case counts."),
    (unique_records, "Only one record per FIPs, per date allowed."),
    (no_nulls_test, "All values are expected to be non-null."),
    (cases_range_test, "Cases must be non-negative and <= 10M."),
    (deaths_range_test, "Deaths must be non-negative and <= 100K.")
]