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])