def test_get_mesh_csvdata_err4(self):
# inconsistent number of levels
fakecsv = StringIO(u"""\
PGA 12.0 42.0 0.14 0.15
PGA 12.0 42.1 0.44 0.45 0.46
PGA 12.0 42.2 0.64
""")
with self.assertRaises(ValueError) as ctx:
readinput.get_mesh_csvdata(
fakecsv, ['PGA'], [2], valid.probabilities)
self.assertIn('Found 3 values, expected 2', str(ctx.exception))
def test_get_mesh_csvdata_err5(self):
# unexpected IMT
fakecsv = StringIO(u"""\
PGA 12.0 42.0 0.14 0.15
PGA 12.0 42.1 0.44 0.45
PGA 12.0 42.2 0.64 0.65
""")
with self.assertRaises(ValueError) as ctx:
readinput.get_mesh_csvdata(
fakecsv, ['PGV'], [3], valid.probabilities)
self.assertIn("Got 'PGA', expected PGV", str(ctx.exception))
def test_get_mesh_csvdata_err5(self):
# unexpected IMT
fakecsv = StringIO(u"""\
PGA 12.0 42.0 0.14 0.15
PGA 12.0 42.1 0.44 0.45
PGA 12.0 42.2 0.64 0.65
""")
with self.assertRaises(ValueError) as ctx:
readinput.get_mesh_csvdata(
fakecsv, ['PGV'], [3], valid.probabilities)
self.assertIn("Got 'PGA', expected PGV", str(ctx.exception))
def test_get_mesh_csvdata_err4(self):
# inconsistent number of levels
fakecsv = StringIO(u"""\
PGA 12.0 42.0 0.14 0.15
PGA 12.0 42.1 0.44 0.45 0.46
PGA 12.0 42.2 0.64
""")
with self.assertRaises(ValueError) as ctx:
readinput.get_mesh_csvdata(
fakecsv, ['PGA'], [2], valid.probabilities)
self.assertIn('Found 3 values, expected 2', str(ctx.exception))
def test_get_mesh_csvdata_err2(self):
# a duplicated point
fakecsv = StringIO(u"""\
PGA 12.0 42.0 0.14 0.15 0.16
PGA 12.0 42.1 0.44 0.45 0.46
PGA 12.0 42.2 0.64 0.65 0.66
PGV 12.0 42.1 0.24 0.25 0.26
PGV 12.0 42.1 0.34 0.35 0.36
""")
with self.assertRaises(readinput.DuplicatedPoint) as ctx:
readinput.get_mesh_csvdata(
fakecsv, ['PGA', 'PGV'], [3, 3], valid.probabilities)
self.assertIn('line 5', str(ctx.exception))
def test_get_mesh_csvdata_err2(self):
# a duplicated point
fakecsv = StringIO(u"""\
PGA 12.0 42.0 0.14 0.15 0.16
PGA 12.0 42.1 0.44 0.45 0.46
PGA 12.0 42.2 0.64 0.65 0.66
PGV 12.0 42.1 0.24 0.25 0.26
PGV 12.0 42.1 0.34 0.35 0.36
""")
with self.assertRaises(readinput.DuplicatedPoint) as ctx:
readinput.get_mesh_csvdata(
fakecsv, ['PGA', 'PGV'], [3, 3], valid.probabilities)
self.assertIn('line 5', str(ctx.exception))
def test_get_mesh_csvdata_err3(self):
# a missing location for PGV
fakecsv = StringIO(u"""\
PGA 12.0 42.0 0.14 0.15 0.16
PGA 12.0 42.1 0.44 0.45 0.46
PGA 12.0 42.2 0.64 0.65 0.66
PGV 12.0 42.0 0.24 0.25 0.26
PGV 12.0 42.1 0.34 0.35 0.36
""")
with self.assertRaises(ValueError) as ctx:
readinput.get_mesh_csvdata(
fakecsv, ['PGA', 'PGV'], [3, 3], valid.probabilities)
self.assertEqual(str(ctx.exception),
'Inconsistent locations between PGA and PGV')
def test_get_mesh_csvdata_err1(self):
# a negative probability
fakecsv = StringIO(u"""\
PGA 12.0 42.0 0.14 0.15 0.16
PGA 12.0 42.1 0.44 0.45 0.46
PGA 12.0 42.2 0.64 0.65 0.66
PGV 12.0 42.0 0.24 0.25 -0.26
PGV 12.0 42.1 0.34 0.35 0.36
PGV 12.0 42.2 0.54 0.55 0.56
""")
with self.assertRaises(ValueError) as ctx:
readinput.get_mesh_csvdata(
fakecsv, ['PGA', 'PGV'], [3, 3], valid.probabilities)
self.assertIn('line 4', str(ctx.exception))
def test_get_mesh_csvdata_err3(self):
# a missing location for PGV
fakecsv = StringIO(u"""\
PGA 12.0 42.0 0.14 0.15 0.16
PGA 12.0 42.1 0.44 0.45 0.46
PGA 12.0 42.2 0.64 0.65 0.66
PGV 12.0 42.0 0.24 0.25 0.26
PGV 12.0 42.1 0.34 0.35 0.36
""")
with self.assertRaises(ValueError) as ctx:
readinput.get_mesh_csvdata(
fakecsv, ['PGA', 'PGV'], [3, 3], valid.probabilities)
self.assertEqual(str(ctx.exception),
'Inconsistent locations between PGA and PGV')
def test_get_mesh_csvdata_err1(self):
# a negative probability
fakecsv = StringIO(u"""\
PGA 12.0 42.0 0.14 0.15 0.16
PGA 12.0 42.1 0.44 0.45 0.46
PGA 12.0 42.2 0.64 0.65 0.66
PGV 12.0 42.0 0.24 0.25 -0.26
PGV 12.0 42.1 0.34 0.35 0.36
PGV 12.0 42.2 0.54 0.55 0.56
""")
with self.assertRaises(ValueError) as ctx:
readinput.get_mesh_csvdata(
fakecsv, ['PGA', 'PGV'], [3, 3], valid.probabilities)
self.assertIn('line 4', str(ctx.exception))
def test_get_mesh_csvdata_different_levels(self):
fakecsv = StringIO(u"""\
PGA 12.0 42.0 0.14 0.15 0.16
PGA 12.0 42.1 0.44 0.45 0.46
PGA 12.0 42.2 0.64 0.65 0.66
PGV 12.0 42.0 0.24 0.25
PGV 12.0 42.1 0.34 0.35
PGV 12.0 42.2 0.54 0.55
""")
mesh, data = readinput.get_mesh_csvdata(fakecsv, ['PGA', 'PGV'],
[3, 2], valid.probabilities)
assert_allclose(mesh.lons, [12., 12., 12.])
assert_allclose(mesh.lats, [42., 42.1, 42.2])
assert_allclose(
data['PGA'],
[[0.14, 0.15, 0.16], [0.44, 0.45, 0.46], [0.64, 0.65, 0.66]])
assert_allclose(data['PGV'],
[[0.24, 0.25], [0.34, 0.35], [0.54, 0.55]])
def test_get_mesh_csvdata_different_levels(self):
fakecsv = StringIO(u"""\
PGA 12.0 42.0 0.14 0.15 0.16
PGA 12.0 42.1 0.44 0.45 0.46
PGA 12.0 42.2 0.64 0.65 0.66
PGV 12.0 42.0 0.24 0.25
PGV 12.0 42.1 0.34 0.35
PGV 12.0 42.2 0.54 0.55
""")
mesh, data = readinput.get_mesh_csvdata(
fakecsv, ['PGA', 'PGV'], [3, 2], valid.probabilities)
assert_allclose(mesh.lons, [12., 12., 12.])
assert_allclose(mesh.lats, [42., 42.1, 42.2])
assert_allclose(data['PGA'], [[0.14, 0.15, 0.16],
[0.44, 0.45, 0.46],
[0.64, 0.65, 0.66]])
assert_allclose(data['PGV'], [[0.24, 0.25],
[0.34, 0.35],
[0.54, 0.55]])
def test_get_mesh_csvdata_ok(self):
fakecsv = StringIO("""\
PGA 12.0 42.0 0.14 0.15 0.16
PGA 12.0 42.1 0.44 0.45 0.46
PGA 12.0 42.2 0.64 0.65 0.66
PGV 12.0 42.0 0.24 0.25 0.26
PGV 12.0 42.1 0.34 0.35 0.36
PGV 12.0 42.2 0.54 0.55 0.56
""")
mesh, data = readinput.get_mesh_csvdata(
fakecsv, ['PGA', 'PGV'], [3, 3], valid.probabilities)
assert_allclose(mesh.lons, [12., 12., 12.])
assert_allclose(mesh.lats, [42., 42.1, 42.2])
assert_allclose(data['PGA'], [[0.14, 0.15, 0.16],
[0.44, 0.45, 0.46],
[0.64, 0.65, 0.66]])
assert_allclose(data['PGV'], [[0.24, 0.25, 0.26],
[0.34, 0.35, 0.36],
[0.54, 0.55, 0.56]])
