def test_none_location(self):
# TODO this isn't about Syria, move it somewhere else
counts = get_filter_counts(self.session, location_ids=['NULL'])
self.assertGreater(len(counts), 1000)
self.assertEqual(
counts, get_filter_counts(self.session, location_ids=['null']))
self.assertEqual(counts,
get_filter_counts(self.session, location_ids=[None]))
counts2 = get_filter_counts(self.session, location_ids=['NULL', 1])
self.assertGreater(len(counts2), len(counts))
def test_filter_counts(self):
f_c = get_filter_counts(self.session,
fromdate=self.start_date,
todate=self.plus_1_yr,
location_ids=self.syria_location_ids)
# print(f_c)
self.assertGreater(len(f_c), 1000)
def test_none_specific_reported_figure_1(self):
# TODO this isn't about Syria, move it somewhere else
counts = get_filter_counts(self.session,
specific_reported_figures=['NULL'])
srf_counts = [
c for c in counts if c['filter_type'] == 'specific_reported_figure'
]
self.assertEqual(len(srf_counts), 1)
self.assertTrue([c for c in srf_counts if c['value'] is None])
self.assertEqual(
counts,
get_filter_counts(self.session,
specific_reported_figures=['null']))
self.assertEqual(
counts,
get_filter_counts(self.session, specific_reported_figures=[None]))
def test_specific_reported_figure(self):
# TODO this isn't about Syria, move it somewhere else
counts = get_filter_counts(self.session,
specific_reported_figures=[1, 1000])
srf_counts = [
c for c in counts if c['filter_type'] == 'specific_reported_figure'
]
self.assertGreater(len(srf_counts), 2)
self.assertFalse([c for c in srf_counts if c['value'] is None])
self.assertTrue([c for c in srf_counts if c['value'] == 1])
def filters():
session = Session()
try:
data = request.get_json(silent=True) or request.form
filters = filter_params(data)
result = get_filter_counts(session, **filters)
resp = jsonify(result)
resp.status_code = 200
return resp
finally:
session.close()
def test_filter_counts_speed(self):
for end_date in (self.plus_1_mo, self.plus_3_mo, self.plus_6_mo):
# Adding this usually fails: , self.plus_1_yr):
t0 = time.time()
f_c = get_filter_counts(self.session,
fromdate=self.start_date,
todate=end_date,
location_ids=self.syria_location_ids)
t1 = time.time()
print('{} - {}: {}s'.format(self.start_date, end_date, t1 - t0))
self.assertLess(
t1 - t0, 1.0,
'Calculating filter counts {} - {} took too long'.format(
self.start_date, end_date))
