def test_store_site_model(self): # Setup site_model = helpers.get_data_path("site_model.xml") exp_site_model = [ dict(lon=-122.5, lat=37.5, vs30=800.0, vs30_type="measured", z1pt0=100.0, z2pt5=5.0), dict(lon=-122.6, lat=37.6, vs30=801.0, vs30_type="measured", z1pt0=101.0, z2pt5=5.1), dict(lon=-122.7, lat=37.7, vs30=802.0, vs30_type="measured", z1pt0=102.0, z2pt5=5.2), dict(lon=-122.8, lat=37.8, vs30=803.0, vs30_type="measured", z1pt0=103.0, z2pt5=5.3), dict(lon=-122.9, lat=37.9, vs30=804.0, vs30_type="measured", z1pt0=104.0, z2pt5=5.4), ] job = models.OqJob.objects.create(user_name="openquake") ids = general.store_site_model(job, site_model) actual_site_model = models.SiteModel.objects.filter(job=job).order_by("id") for i, exp in enumerate(exp_site_model): act = actual_site_model[i] self.assertAlmostEqual(exp["lon"], act.location.x) self.assertAlmostEqual(exp["lat"], act.location.y) self.assertAlmostEqual(exp["vs30"], act.vs30) self.assertEqual(exp["vs30_type"], act.vs30_type) self.assertAlmostEqual(exp["z1pt0"], act.z1pt0) self.assertAlmostEqual(exp["z2pt5"], act.z2pt5) # last, check that the `store_site_model` function returns all of the # newly-inserted records for i, s in enumerate(ids): self.assertEqual(s, actual_site_model[i].id)
def test_store_site_model(self): # Setup site_model = helpers.get_data_path('site_model.xml') exp_site_model = [ dict(lon=-122.5, lat=37.5, vs30=800.0, vs30_type="measured", z1pt0=100.0, z2pt5=5.0), dict(lon=-122.6, lat=37.6, vs30=801.0, vs30_type="measured", z1pt0=101.0, z2pt5=5.1), dict(lon=-122.7, lat=37.7, vs30=802.0, vs30_type="measured", z1pt0=102.0, z2pt5=5.2), dict(lon=-122.8, lat=37.8, vs30=803.0, vs30_type="measured", z1pt0=103.0, z2pt5=5.3), dict(lon=-122.9, lat=37.9, vs30=804.0, vs30_type="measured", z1pt0=104.0, z2pt5=5.4), ] job = models.OqJob.objects.create(user_name="openquake") ids = general.store_site_model(job, site_model) actual_site_model = models.SiteModel.objects.filter( job=job).order_by('id') for i, exp in enumerate(exp_site_model): act = actual_site_model[i] self.assertAlmostEqual(exp['lon'], act.location.x) self.assertAlmostEqual(exp['lat'], act.location.y) self.assertAlmostEqual(exp['vs30'], act.vs30) self.assertEqual(exp['vs30_type'], act.vs30_type) self.assertAlmostEqual(exp['z1pt0'], act.z1pt0) self.assertAlmostEqual(exp['z2pt5'], act.z2pt5) # last, check that the `store_site_model` function returns all of the # newly-inserted records for i, s in enumerate(ids): self.assertEqual(s, actual_site_model[i].id)
def test_store_site_model(self): # Setup inp = models.Input( owner=models.OqUser.objects.get(id=1), path='fake_path', digest='fake_digest', input_type='site_model', size=0) inp.save() site_model = helpers.get_data_path('site_model.xml') exp_site_model = [ dict(lon=-122.5, lat=37.5, vs30=800.0, vs30_type="measured", z1pt0=100.0, z2pt5=5.0), dict(lon=-122.6, lat=37.6, vs30=801.0, vs30_type="measured", z1pt0=101.0, z2pt5=5.1), dict(lon=-122.7, lat=37.7, vs30=802.0, vs30_type="measured", z1pt0=102.0, z2pt5=5.2), dict(lon=-122.8, lat=37.8, vs30=803.0, vs30_type="measured", z1pt0=103.0, z2pt5=5.3), dict(lon=-122.9, lat=37.9, vs30=804.0, vs30_type="measured", z1pt0=104.0, z2pt5=5.4), ] ret_val = general.store_site_model(inp, site_model) actual_site_model = models.SiteModel.objects.filter( input=inp.id).order_by('id') for i, exp in enumerate(exp_site_model): act = actual_site_model[i] self.assertAlmostEqual(exp['lon'], act.location.x) self.assertAlmostEqual(exp['lat'], act.location.y) self.assertAlmostEqual(exp['vs30'], act.vs30) self.assertEqual(exp['vs30_type'], act.vs30_type) self.assertAlmostEqual(exp['z1pt0'], act.z1pt0) self.assertAlmostEqual(exp['z2pt5'], act.z2pt5) # last, check that the `store_site_model` function returns all of the # newly-inserted records # an `equals` check just compares the ids for i, val in enumerate(ret_val): self.assertEqual(val, actual_site_model[i])