def test_generate_erf_returns_erf_via_kvs(self):
results = []
result_keys = []
expected_values = {}
print kvs.tokens.ERF_KEY_TOKEN
for job_id in TASK_JOBID_SIMPLE:
erf_key = kvs.generate_product_key(job_id,
kvs.tokens.ERF_KEY_TOKEN)
# Build the expected values
expected_values[erf_key] = json.JSONEncoder().encode([job_id])
# Get our result keys
result_keys.append(erf_key)
# Spawn our tasks.
results.append(tasks.generate_erf.apply_async(args=[job_id]))
test.wait_for_celery_tasks(results)
result_values = self.kvs_client.get_multi(result_keys)
self.assertEqual(result_values, expected_values)
def test_compute_hazard_curve_all_sites(self):
results = []
block_id = 8801
for job_id in TASK_JOBID_SIMPLE:
self._prepopulate_sites_for_block(job_id, block_id)
results.append(tasks.compute_hazard_curve.apply_async(
args=[job_id, block_id]))
test.wait_for_celery_tasks(results)
for result in results:
for res in result.get():
self.assertEqual(res, ONE_CURVE_MODEL)
def test_compute_mgm_intensity(self):
results = []
block_id = 8801
site = "Testville,TestLand"
mgm_intensity = json.JSONDecoder().decode(MEAN_GROUND_INTENSITY)
for job_id in TASK_JOBID_SIMPLE:
mgm_key = kvs.generate_product_key(job_id,
kvs.tokens.MGM_KEY_TOKEN, block_id, site)
self.kvs_client.set(mgm_key, MEAN_GROUND_INTENSITY)
results.append(tasks.compute_mgm_intensity.apply_async(
args=[job_id, block_id, site]))
test.wait_for_celery_tasks(results)
for result in results:
self.assertEqual(mgm_intensity, result.get())
