def test_cut_window(self):
testRFCut = RiverFlood()
centr = RiverFlood.select_window_area(['AUT'])
testRFCut.centroids.lon = centr.lon
testRFCut.centroids.lat = centr.lat
lon = np.arange(7, 20, 0.2)
lat = np.arange(40, 50, 0.2)
test_window = [[4, 24], [55, 45]]
self.assertTrue(np.array_equal(testRFCut._cut_window(lon, lat),
test_window))
def test_wrong_iso3_fail(self):
emptyFlood = RiverFlood()
with self.assertRaises(KeyError):
RiverFlood.select_exact_area(['OYY'])
with self.assertRaises(KeyError):
RiverFlood.select_window_area(['OYY'])
with self.assertRaises(AttributeError):
emptyFlood.set_from_nc(years=[2600])
with self.assertRaises(KeyError):
emptyFlood.set_from_nc(reg=['OYY'])
def test_set_centroids_from_file(self):
testRFCentr = RiverFlood()
lon = [1, 2, 3]
lat = [1, 2, 3]
testRFCentr._set_centroids_from_file(lon, lat)
test_centroids_lon = np.array([1, 2, 3, 1, 2, 3, 1, 2, 3])
test_centroids_lat = np.array([1, 1, 1, 2, 2, 2, 3, 3, 3])
self.assertTrue(np.array_equal(testRFCentr.centroids.lon,
test_centroids_lon))
self.assertTrue(np.array_equal(testRFCentr.centroids.lat,
test_centroids_lat))
def test_select_events(self):
testRFTime = RiverFlood()
tt1 = dt.datetime.strptime('1988-07-02', '%Y-%m-%d')
tt2 = dt.datetime.strptime('2010-04-01', '%Y-%m-%d')
tt3 = dt.datetime.strptime('1997-07-02', '%Y-%m-%d')
tt4 = dt.datetime.strptime('1990-07-02', '%Y-%m-%d')
years = [2010, 1997]
test_time = np.array([tt1, tt2, tt3, tt4])
self.assertTrue(
np.array_equal(testRFTime._select_event(test_time, years), [1, 2]))
years = [1988, 1990]
self.assertTrue(
np.array_equal(testRFTime._select_event(test_time, years), [0, 3]))
def test_select_window_area(self):
testWinCentroids = RiverFlood.select_window_area(['DEU'])
self.assertEqual(testWinCentroids.lon.shape[0], 57505)
self.assertAlmostEqual(np.max(testWinCentroids.lon),
15.979019799999975)
self.assertAlmostEqual(np.min(testWinCentroids.lon),
4.9790373999999815)
self.assertAlmostEqual(np.max(testWinCentroids.lat),
55.97889979999998)
self.assertAlmostEqual(np.min(testWinCentroids.lat),
46.97891419999998)
self.assertAlmostEqual(testWinCentroids.lon[1000],
13.520690399999978)
self.assertAlmostEqual(testWinCentroids.lat[1000],
47.10391399999999)
self.assertAlmostEqual(testWinCentroids.lon[2000],
11.020694399999968)
self.assertAlmostEqual(testWinCentroids.lat[2000],
47.270580399999986)
self.assertAlmostEqual(testWinCentroids.lon[3000],
8.520698399999986)
self.assertAlmostEqual(testWinCentroids.lat[3000],
47.43724679999998)
self.assertEqual(testWinCentroids.id[0], 0)
self.assertEqual(testWinCentroids.id[1204], 1204)
self.assertEqual(testWinCentroids.id[57504], 57504)
def test_exact_area_selection(self):
testCentroids = RiverFlood.select_exact_area(['LIE'])
self.assertEqual(testCentroids.lon.shape[0], 13)
self.assertAlmostEqual(testCentroids.lon[0], 9.5206968)
self.assertAlmostEqual(testCentroids.lon[1], 9.5623634)
self.assertAlmostEqual(testCentroids.lon[2], 9.60403)
self.assertAlmostEqual(testCentroids.lon[3], 9.5206968)
self.assertAlmostEqual(testCentroids.lon[4], 9.5623634)
self.assertAlmostEqual(testCentroids.lon[5], 9.60403)
self.assertAlmostEqual(testCentroids.lon[6], 9.5206968)
self.assertAlmostEqual(testCentroids.lon[7], 9.5623634)
self.assertAlmostEqual(testCentroids.lon[8], 9.60403)
self.assertAlmostEqual(testCentroids.lon[9], 9.5206968)
self.assertAlmostEqual(testCentroids.lon[10], 9.5623634)
self.assertAlmostEqual(testCentroids.lon[11], 9.5206968)
self.assertAlmostEqual(testCentroids.lon[12], 9.5623634)
self.assertAlmostEqual(testCentroids.lat[0], 47.0622474)
self.assertAlmostEqual(testCentroids.lat[1], 47.0622474)
self.assertAlmostEqual(testCentroids.lat[2], 47.0622474)
self.assertAlmostEqual(testCentroids.lat[3], 47.103914)
self.assertAlmostEqual(testCentroids.lat[4], 47.103914)
self.assertAlmostEqual(testCentroids.lat[5], 47.103914)
self.assertAlmostEqual(testCentroids.lat[6], 47.1455806)
self.assertAlmostEqual(testCentroids.lat[7], 47.1455806)
self.assertAlmostEqual(testCentroids.lat[8], 47.1455806)
self.assertAlmostEqual(testCentroids.lat[9], 47.1872472)
self.assertAlmostEqual(testCentroids.lat[10], 47.1872472)
self.assertAlmostEqual(testCentroids.lat[11], 47.2289138)
self.assertAlmostEqual(testCentroids.lat[12], 47.2289138)
self.assertEqual(testCentroids.id[0], 0)
self.assertEqual(testCentroids.id[5], 5)
self.assertEqual(testCentroids.id[12], 12)
def test_select_model_run(self):
testRFModel = RiverFlood()
flood_dir = '/home/test/flood/'
rf_model = 'LPJmL'
cl_model = 'wfdei'
prot_std = 'flopros'
scenario = 'historical'
self.assertEqual(testRFModel._select_model_run(flood_dir, rf_model,
cl_model, scenario,
prot_std)[0],
'/home/test/flood/flddph_LPJmL_wfdei_' +
'flopros_gev_0.1.nc')
self.assertEqual(testRFModel._select_model_run(flood_dir, rf_model,
cl_model, scenario,
prot_std, proj=True)[0],
'/home/test/flood/flddph_LPJmL_wfdei_' +
'historical_flopros_gev_picontrol_2000_0.1.nc')
def test_exact_area_selection(self):
testCentroids = RiverFlood.select_exact_area(['LIE'])
self.assertEqual(testCentroids.lon.shape[0], 13)
self.assertAlmostEqual(testCentroids.lon[0], 9.5206968)
self.assertAlmostEqual(testCentroids.lon[1], 9.5623634)
self.assertAlmostEqual(testCentroids.lon[2], 9.60403)
self.assertAlmostEqual(testCentroids.lon[3], 9.5206968)
self.assertAlmostEqual(testCentroids.lon[4], 9.5623634)
self.assertAlmostEqual(testCentroids.lon[5], 9.60403)
self.assertAlmostEqual(testCentroids.lon[6], 9.5206968)
self.assertAlmostEqual(testCentroids.lon[7], 9.5623634)
self.assertAlmostEqual(testCentroids.lon[8], 9.60403)
self.assertAlmostEqual(testCentroids.lon[9], 9.5206968)
self.assertAlmostEqual(testCentroids.lon[10], 9.5623634)
self.assertAlmostEqual(testCentroids.lon[11], 9.5206968)
self.assertAlmostEqual(testCentroids.lon[12], 9.5623634)
self.assertAlmostEqual(testCentroids.lat[0], 47.0622474)
self.assertAlmostEqual(testCentroids.lat[1], 47.0622474)
self.assertAlmostEqual(testCentroids.lat[2], 47.0622474)
self.assertAlmostEqual(testCentroids.lat[3], 47.103914)
self.assertAlmostEqual(testCentroids.lat[4], 47.103914)
self.assertAlmostEqual(testCentroids.lat[5], 47.103914)
self.assertAlmostEqual(testCentroids.lat[6], 47.1455806)
self.assertAlmostEqual(testCentroids.lat[7], 47.1455806)
self.assertAlmostEqual(testCentroids.lat[8], 47.1455806)
self.assertAlmostEqual(testCentroids.lat[9], 47.1872472)
self.assertAlmostEqual(testCentroids.lat[10], 47.1872472)
self.assertAlmostEqual(testCentroids.lat[11], 47.2289138)
self.assertAlmostEqual(testCentroids.lat[12], 47.2289138)
self.assertEqual(testCentroids.id[0], 0)
self.assertEqual(testCentroids.id[5], 5)
self.assertEqual(testCentroids.id[12], 12)
testCentroids = RiverFlood.select_exact_area(['DEU'])
self.assertAlmostEqual(np.max(testCentroids.lon), 15.020687999999979)
self.assertAlmostEqual(np.min(testCentroids.lon), 5.895702599999964)
self.assertAlmostEqual(np.max(testCentroids.lat), 55.0622346)
self.assertAlmostEqual(np.min(testCentroids.lat), 47.312247)
def test_full_flood(self):
""" read_flood"""
testRF = RiverFlood()
testRF.set_from_nc()
self.assertEqual(testRF.centroids.lat.shape[0], 37324800)
self.assertAlmostEqual(np.max(testRF.centroids.lon),
179.97916666666663)
self.assertAlmostEqual(np.min(testRF.centroids.lon),
-179.97916666666666)
self.assertAlmostEqual(np.max(testRF.centroids.lat),
89.97916666666667)
self.assertAlmostEqual(np.min(testRF.centroids.lat),
-89.97916666666666)
self.assertAlmostEqual(testRF.centroids.lon[20000000],
113.35416666666663)
self.assertAlmostEqual(testRF.centroids.lat[20000000],
-6.4375)
self.assertAlmostEqual(testRF.centroids.lon[10000000],
-33.3125)
self.assertAlmostEqual(testRF.centroids.lat[10000000],
41.770833333333336)
self.assertEqual(testRF.orig[0], 1)
self.assertEqual(np.argmax(testRF.intensity), 26190437)
self.assertAlmostEqual(np.max(testRF.fraction), 1.0)
self.assertAlmostEqual(testRF.intensity[0, 3341441], 0.9583404)
self.assertAlmostEqual(testRF.intensity[0, 3341442], 0.9583404)
self.assertAlmostEqual(testRF.fraction[0, 3341441], 0.41666666)
self.assertAlmostEqual(testRF.fraction[0, 3341442], 0.375)
self.assertEqual(np.argmax(testRF.fraction[0]), 3341440)
testRFReg = RiverFlood()
testRFReg.set_from_nc(reg='SWA')
self.assertEqual(testRFReg.centroids.lat.shape[0], 301181)
self.assertAlmostEqual(np.max(testRFReg.centroids.lon),
101.14555019999997)
self.assertAlmostEqual(np.min(testRFReg.centroids.lon),
60.52061519999998)
self.assertAlmostEqual(np.max(testRFReg.centroids.lat),
38.43726119999998)
self.assertAlmostEqual(np.min(testRFReg.centroids.lat),
-0.6876762000000127)
self.assertEqual(testRFReg.date[0], 730303)
self.assertEqual((date.fromordinal(testRFReg.date[0]).year), 2000)
self.assertEqual(testRFReg.orig[0], 1)
self.assertAlmostEqual(np.max(testRFReg.intensity), 16.69780921936035)
self.assertEqual(np.argmax(testRFReg.intensity), 40613)
self.assertEqual(np.argmax(testRFReg.fraction), 126135)
def test_cut_flooded_area(self):
testRFwin = RiverFlood()
winAFG = RiverFlood.select_window_area(['AFG'])
testRFwin.set_from_nc(centroids=winAFG)
afg = RiverFlood.select_exact_area(['AFG'])
testRFwin.set_flooded_area_cut(afg.coord)
self.assertAlmostEqual(np.max(testRFwin.fla_ann_centr[0]),
17200498.22927546)
self.assertEqual(np.argmax(testRFwin.fla_ann_centr[0]),
32610)
self.assertAlmostEqual(np.max(testRFwin.fla_ev_centr[0]),
17200498.22927546)
self.assertEqual(np.argmax(testRFwin.fla_ev_centr[0]),
32610)
self.assertAlmostEqual(testRFwin.fla_event[0],
6244242013.5826435, 4)
self.assertAlmostEqual(testRFwin.fla_annual[0],
6244242013.5826435, 4)
self.assertAlmostEqual(testRFwin.fla_ann_av,
6244242013.5826435, 4)
self.assertAlmostEqual(testRFwin.fla_ev_av,
6244242013.5826435, 4)
def test_flooded_area(self):
testRFArea = RiverFlood()
dph_path = HAZ_DEMO_FLDDPH
frc_path = HAZ_DEMO_FLDFRC
testRFArea.set_from_nc(countries=['AUT'], dph_path=dph_path,
frc_path=frc_path)
self.assertEqual(testRFArea.centroids.lat.shape[0], 5782)
self.assertAlmostEqual(np.max(testRFArea.centroids.lon),
17.104017999999968)
self.assertAlmostEqual(np.min(testRFArea.centroids.lon),
9.562363399999981)
self.assertAlmostEqual(np.max(testRFArea.centroids.lat),
48.978911)
self.assertAlmostEqual(np.min(testRFArea.centroids.lat),
46.39558179999999)
self.assertEqual(testRFArea.orig[0], 1)
self.assertAlmostEqual(np.max(testRFArea.intensity),
9.613386154174805)
self.assertEqual(np.argmax(testRFArea.intensity), 2786)
self.assertAlmostEqual(np.max(testRFArea.fraction),
0.5103999972343445)
self.assertEqual(np.argmax(testRFArea.fraction), 3391)
testRFArea.set_flooded_area()
self.assertAlmostEqual(np.max(testRFArea.fla_ev_centr),
7396511.421906647)
self.assertEqual(np.argmax(testRFArea.fla_ev_centr), 3391)
self.assertAlmostEqual(np.max(testRFArea.fla_ann_centr),
7396511.421906647)
self.assertEqual(np.argmax(testRFArea.fla_ann_centr),
3391)
self.assertAlmostEqual(testRFArea.fla_event[0],
229956891.5531019, 5)
self.assertAlmostEqual(testRFArea.fla_annual[0],
229956891.5531019, 5)
testRFset = RiverFlood()
testRFset.set_from_nc(countries=['AFG'])
years = [2000, 2001, 2002]
manipulated_dates = [730303, 730669, 731034]
for i in range(len(years)):
testRFaddset = RiverFlood()
testRFaddset.set_from_nc(countries=['AFG'])
testRFaddset.date = [manipulated_dates[i]]
if i == 0:
testRFaddset.event_name = ['2000_2']
else:
testRFaddset.event_name = [str(years[i])]
testRFset.append(testRFaddset)
testRFset.set_flooded_area()
self.assertEqual(testRFset.fla_event.shape[0], 4)
self.assertEqual(testRFset.fla_annual.shape[0], 3)
self.assertAlmostEqual(np.max(testRFset.fla_ev_centr[0]),
17200498.22927546)
self.assertEqual(np.argmax(testRFset.fla_ev_centr[0]),
32610)
self.assertAlmostEqual(np.max(testRFset.fla_ev_centr[2]),
17200498.22927546)
self.assertEqual(np.argmax(testRFset.fla_ev_centr[2]),
32610)
self.assertAlmostEqual(np.max(testRFset.fla_ann_centr[0]),
34400996.45855092)
self.assertEqual(np.argmax(testRFset.fla_ann_centr[0]),
32610)
self.assertAlmostEqual(np.max(testRFset.fla_ann_centr[2]),
17200498.22927546)
self.assertEqual(np.argmax(testRFset.fla_ann_centr[2]),
32610)
self.assertAlmostEqual(testRFset.fla_event[0],
6244242013.5826435, 4)
self.assertAlmostEqual(testRFset.fla_annual[0],
12488484027.165287, 3)
self.assertAlmostEqual(testRFset.fla_ann_av,
8325656018.110191, 4)
self.assertAlmostEqual(testRFset.fla_ev_av,
6244242013.5826435, 4)
def test_flooded_area(self):
dph_path = HAZ_DEMO_FLDDPH
frc_path = HAZ_DEMO_FLDFRC
testRFset = RiverFlood()
testRFset.set_from_nc(countries=['AFG'],
dph_path=dph_path,
frc_path=frc_path)
years = [2000, 2001, 2002]
manipulated_dates = [730303, 730669, 731034]
for i in range(len(years)):
testRFaddset = RiverFlood()
testRFaddset.set_from_nc(countries=['AFG'])
testRFaddset.date = [manipulated_dates[i]]
if i == 0:
testRFaddset.event_name = ['2000_2']
else:
testRFaddset.event_name = [str(years[i])]
testRFset.append(testRFaddset)
testRFset.set_flooded_area()
self.assertEqual(testRFset.units, 'm')
self.assertEqual(testRFset.fla_event.shape[0], 4)
self.assertEqual(testRFset.fla_annual.shape[0], 3)
self.assertAlmostEqual(np.max(testRFset.fla_ev_centr[0]),
17200498.22927546)
self.assertEqual(np.argmax(testRFset.fla_ev_centr[0]), 32610)
self.assertAlmostEqual(np.max(testRFset.fla_ev_centr[2]),
17200498.22927546)
self.assertEqual(np.argmax(testRFset.fla_ev_centr[2]), 32610)
self.assertAlmostEqual(np.max(testRFset.fla_ann_centr[0]),
34400996.45855092)
self.assertEqual(np.argmax(testRFset.fla_ann_centr[0]), 32610)
self.assertAlmostEqual(np.max(testRFset.fla_ann_centr[2]),
17200498.22927546)
self.assertEqual(np.argmax(testRFset.fla_ann_centr[2]), 32610)
self.assertAlmostEqual(testRFset.fla_event[0], 6244242013.5826435, 4)
self.assertAlmostEqual(testRFset.fla_annual[0], 12488484027.165287, 3)
self.assertAlmostEqual(testRFset.fla_ann_av, 8325656018.110191, 4)
self.assertAlmostEqual(testRFset.fla_ev_av, 6244242013.5826435, 4)