def test_equal_same(self):
"""Append the same hazard. Obtain initial hazard."""
haz1 = Hazard('TC')
haz1.read_excel(HAZ_TEMPLATE_XLS)
haz2 = Hazard('TC')
haz2.read_excel(HAZ_TEMPLATE_XLS)
haz1.append(haz2)
haz1.remove_duplicates()
haz1.check()
self.assertEqual(haz1.event_name, haz2.event_name)
self.assertTrue(np.array_equal(haz1.event_id, haz2.event_id))
self.assertTrue(np.array_equal(haz1.frequency, haz2.frequency))
self.assertTrue(np.array_equal(haz1.date, haz2.date))
self.assertTrue(np.array_equal(haz1.orig, haz2.orig))
self.assertTrue(
np.array_equal(haz1.intensity.todense(), haz2.intensity.todense()))
self.assertTrue(
np.array_equal(haz1.fraction.todense(), haz2.fraction.todense()))
self.assertTrue((haz1.intensity != haz2.intensity).nnz == 0)
self.assertTrue((haz1.fraction != haz2.fraction).nnz == 0)
self.assertEqual(haz1.units, haz2.units)
self.assertEqual(haz1.tag.file_name,
[haz2.tag.file_name, haz2.tag.file_name])
self.assertEqual(haz1.tag.haz_type, haz2.tag.haz_type)
self.assertEqual(haz1.tag.description,
[haz2.tag.description, haz2.tag.description])
def test_append_new_var_pass(self):
""" New variable appears if hazard to append is empty. """
haz = dummy_hazard()
haz.new_var = np.ones(haz.size)
app_haz = Hazard('TC')
app_haz.append(haz)
self.assertIn('new_var', app_haz.__dict__)
def test_append_to_empty_fill(self):
"""Append to an empty hazard a filled one. Obtain filled one."""
haz1 = Hazard('TC')
haz2 = Hazard('TC')
haz2.read_excel(HAZ_TEMPLATE_XLS)
haz1.append(haz2)
haz1.check()
# expected values
haz1_orig = Hazard('TC')
haz1_orig.read_excel(HAZ_TEMPLATE_XLS)
self.assertEqual(haz1.event_name, haz1_orig.event_name)
self.assertTrue(np.array_equal(haz1.event_id, haz1_orig.event_id))
self.assertTrue(np.array_equal(haz1.frequency, haz1_orig.frequency))
self.assertTrue(np.array_equal(haz1.date, haz1_orig.date))
self.assertTrue(np.array_equal(haz1.orig, haz1_orig.orig))
self.assertTrue((haz1.intensity != haz1_orig.intensity).nnz == 0)
self.assertTrue((haz1.fraction != haz1_orig.fraction).nnz == 0)
self.assertEqual(haz1.units, haz1_orig.units)
self.assertEqual(haz1.tag.file_name, haz1_orig.tag.file_name)
self.assertEqual(haz1.tag.haz_type, haz1_orig.tag.haz_type)
self.assertEqual(haz1.tag.description, haz1_orig.tag.description)
def test_append_empty_fill(self):
"""Append an empty. Obtain initial hazard."""
haz1 = Hazard('TC')
haz1.read_excel(HAZ_TEMPLATE_XLS)
haz2 = Hazard('TC')
haz2.centroids.geometry.crs = {'init':'epsg:4326'}
haz1.append(haz2)
haz1.check()
# expected values
haz1_orig = Hazard('TC')
haz1_orig.read_excel(HAZ_TEMPLATE_XLS)
self.assertEqual(haz1.event_name, haz1_orig.event_name)
self.assertTrue(np.array_equal(haz1.event_id, haz1_orig.event_id))
self.assertTrue(np.array_equal(haz1.date, haz1_orig.date))
self.assertTrue(np.array_equal(haz1.orig, haz1_orig.orig))
self.assertTrue(np.array_equal(haz1.frequency, haz1_orig.frequency))
self.assertTrue((haz1.intensity != haz1_orig.intensity).nnz == 0)
self.assertTrue((haz1.fraction != haz1_orig.fraction).nnz == 0)
self.assertEqual(haz1.units, haz1_orig.units)
self.assertEqual(haz1.tag.file_name, haz1_orig.tag.file_name)
self.assertEqual(haz1.tag.haz_type, haz1_orig.tag.haz_type)
self.assertEqual(haz1.tag.description, haz1_orig.tag.description)
