def test_count_calls_returns_correctly_for_police_responses(self):
def get_date_days_ago(days):
return datetime.datetime.now(pytz.utc) - datetime.timedelta(days=days)
incidents = [PoliceIncidentFactory(standardized_address="123 MAIN ST",
call_datetime=get_date_days_ago(5))
for i in range(0, 5)]
incident_tuples = [(incident.standardized_address, incident.call_datetime)
for incident in incidents]
counts = count_calls(incident_tuples, 'call_datetime', 'police_counts', [7, 14])
assert '123 MAIN ST' in counts
assert 'police_counts' in counts['123 MAIN ST']
assert 7 in counts['123 MAIN ST']['police_counts']
assert counts['123 MAIN ST']['police_counts'][7] == 5
assert counts['123 MAIN ST']['police_counts'][14] == 5
def test_count_calls_returns_correctly_for_police_responses(self):
def get_date_days_ago(days):
return datetime.datetime.now(pytz.utc) - datetime.timedelta(days=days)
incidents = [StandardizedPoliceIncidentFactory(standardized_address="123 MAIN ST",
call_datetime=get_date_days_ago(5))
for i in range(0, 5)]
incident_tuples = [(incident.standardized_address, incident.call_datetime)
for incident in incidents]
counts = count_calls(incident_tuples, 'call_datetime', 'police_counts', [7, 14])
assert '123 MAIN ST' in counts
assert 'police_counts' in counts['123 MAIN ST']
assert 7 in counts['123 MAIN ST']['police_counts']
assert counts['123 MAIN ST']['police_counts'][7] == 5
assert counts['123 MAIN ST']['police_counts'][14] == 5
def test_count_calls_returns_empty_when_given_no_incidents(self):
counts = count_calls([], 'alarm_datetime', 'fire_counts', [7, 14])
assert counts == {}
def test_count_calls_returns_empty_when_given_no_incidents(self):
counts = count_calls([], 'alarm_datetime', 'fire_counts', [7, 14])
assert counts == {}