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))
