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])) helpers.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])) helpers.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])) helpers.wait_for_celery_tasks(results) for result in results: self.assertEqual(mgm_intensity, result.get())
def test_compute_mgm_intensity(self): results = [] block_id = 8801 site = "Testville,TestLand" mgm_intensity = json.JSONDecoder().decode(MEAN_GROUND_INTENSITY) job_ids = [helpers.job_from_file(TEST_JOB_FILE).job_id for _ in xrange(4)] for job_id in job_ids: mgm_key = tokens.mgm_key(job_id, 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])) helpers.wait_for_celery_tasks(results) for result in results: self.assertEqual(mgm_intensity, result.get())
def test_compute_mgm_intensity(self): results = [] block_id = 8801 site = "Testville,TestLand" mgm_intensity = json.JSONDecoder().decode(MEAN_GROUND_INTENSITY) job_ids = [ helpers.job_from_file(TEST_JOB_FILE).job_id for _ in xrange(4) ] for job_id in job_ids: mgm_key = tokens.mgm_key(job_id, block_id, site) self.kvs_client.set(mgm_key, MEAN_GROUND_INTENSITY) results.append( classical.compute_mgm_intensity.apply_async( args=[job_id, block_id, site])) helpers.wait_for_celery_tasks(results) for result in results: self.assertEqual(mgm_intensity, result.get())
def test_generate_erf_returns_erf_via_kvs(self): results = [] result_keys = [] expected_values = {} job_ids = [helpers.job_from_file(TEST_JOB_FILE).job_id for _ in xrange(4)] for job_id in job_ids: erf_key = tokens.erf_key(job_id) # 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])) helpers.wait_for_celery_tasks(results) result_values = self.kvs_client.get_multi(result_keys) self.assertEqual(result_values, expected_values)