def _count_combinations(output): actual_combination_counts = {} for key in expected_combination_counts.keys(): actual_combination_counts[key] = 0 for result in output: combination, _ = result combination_dict = json.loads(combination) metric_type = combination_dict.get('metric_type') if metric_type == ProgramMetricType.REFERRAL.value: actual_combination_counts['referrals'] = \ actual_combination_counts['referrals'] + 1 for key in expected_combination_counts: if expected_combination_counts[key] != \ actual_combination_counts[key]: raise BeamAssertException('Failed assert. Count does not' 'match expected value.')
def _count_combinations(output): actual_combination_counts = {} for key in expected_combination_counts.keys(): actual_combination_counts[key] = 0 for result in output: combination_dict, _ = result metric_type = combination_dict.get("metric_type") if metric_type == IncarcerationMetricType.INCARCERATION_ADMISSION: actual_combination_counts["admissions"] = ( actual_combination_counts["admissions"] + 1) elif metric_type == IncarcerationMetricType.INCARCERATION_RELEASE: actual_combination_counts["releases"] = ( actual_combination_counts["releases"] + 1) for key in expected_combination_counts: if expected_combination_counts[ key] != actual_combination_counts[key]: raise BeamAssertException("Failed assert. Count does not" "match expected value.")
def _sum_verifier(init_size, data_size, x): expected = sum(range(data_size)) * init_size if x != expected: raise BeamAssertException( "incorrect sum: expected(%d) actual(%d)" % (expected, x)) return []
def assertEqual(l, r): if l != r: raise BeamAssertException('Assertion failed: %s == %s' % (l, r))
def _count_equal(actual_list): actual_count = len(actual_list) if expected_count != actual_count: raise BeamAssertException('Failed assert: %d == %d' % (expected_count, actual_count))
def _count_equal(actual_list): actual_count = len(actual_list) if expected_count != actual_count: raise BeamAssertException('Expected %d not equal actual %d' % (expected_count, actual_count))
def _items_equal(actual): sorted_expected = sorted(expected) sorted_actual = sorted(actual) if sorted_expected != sorted_actual: raise BeamAssertException('Failed assert: %r != %r' % (sorted_expected, sorted_actual))
def _in_between(actual): _, _, estimate = actual[0] if estimate < lower or estimate > upper: raise BeamAssertException( 'Failed assert: %f not in [%f, %f]' % (estimate, lower, upper))
def assert_true(exp): if not exp: raise BeamAssertException('%s Failed assert True' % repr(exp))