def test_apply_criterion_track(self):
"""Test _apply_criterion function."""
tc = TropCyclone()
tc.intensity = np.zeros((4, 10))
tc.intensity[0, :] = np.arange(10)
tc.intensity[1, 5] = 10
tc.intensity[2, :] = np.arange(10, 20)
tc.intensity[3, 3] = 3
tc.intensity = sparse.csr_matrix(tc.intensity)
tc.basin = ['NA'] * 4
tc.basin[3] = 'NO'
tc.category = np.array([2, 0, 4, 1])
tc.event_id = np.arange(4)
tc.frequency = np.ones(4) * 0.5
tc_cc = tc.set_climate_scenario_knu(ref_year=2050, rcp_scenario=45)
self.assertTrue(np.allclose(tc.intensity[1, :].toarray(), tc_cc.intensity[1, :].toarray()))
self.assertTrue(np.allclose(tc.intensity[3, :].toarray(), tc_cc.intensity[3, :].toarray()))
self.assertFalse(
np.allclose(tc.intensity[0, :].toarray(), tc_cc.intensity[0, :].toarray()))
self.assertFalse(
np.allclose(tc.intensity[2, :].toarray(), tc_cc.intensity[2, :].toarray()))
self.assertTrue(np.allclose(tc.frequency, tc_cc.frequency))
self.assertEqual(
tc_cc.tag.description,
'climate change scenario for year 2050 and RCP 45 from Knutson et al 2015.')
def test_two_criterion_track(self):
"""Test _apply_criterion function with two criteria"""
criterion = [
{'basin': 'NA', 'category': [1, 2, 3, 4, 5],
'year': 2100, 'change': 1.045, 'variable': 'intensity'},
{'basin': 'WP', 'category': [1, 2, 3, 4, 5],
'year': 2100, 'change': 1.025, 'variable': 'intensity'},
{'basin': 'WP', 'category': [1, 2, 3, 4, 5],
'year': 2100, 'change': 1.025, 'variable': 'frequency'},
{'basin': 'NA', 'category': [0, 1, 2, 3, 4, 5],
'year': 2100, 'change': 0.7, 'variable': 'frequency'},
{'basin': 'NA', 'category': [1, 2, 3, 4, 5],
'year': 2100, 'change': 1, 'variable': 'frequency'},
{'basin': 'NA', 'category': [3, 4, 5],
'year': 2100, 'change': 1, 'variable': 'frequency'},
{'basin': 'NA', 'category': [4, 5],
'year': 2100, 'change': 2, 'variable': 'frequency'}
]
scale = 0.75
tc = TropCyclone()
tc.intensity = np.zeros((4, 10))
tc.intensity[0, :] = np.arange(10)
tc.intensity[1, 5] = 10
tc.intensity[2, :] = np.arange(10, 20)
tc.intensity[3, 3] = 3
tc.intensity = sparse.csr_matrix(tc.intensity)
tc.frequency = np.ones(4) * 0.5
tc.basin = ['NA'] * 4
tc.basin[3] = 'WP'
tc.category = np.array([2, 0, 4, 1])
tc.event_id = np.arange(4)
tc_cc = tc._apply_knutson_criterion(criterion, scale)
self.assertTrue(np.allclose(tc.intensity[1, :].toarray(), tc_cc.intensity[1, :].toarray()))
self.assertFalse(
np.allclose(tc.intensity[3, :].toarray(), tc_cc.intensity[3, :].toarray()))
self.assertFalse(
np.allclose(tc.intensity[0, :].toarray(), tc_cc.intensity[0, :].toarray()))
self.assertFalse(
np.allclose(tc.intensity[2, :].toarray(), tc_cc.intensity[2, :].toarray()))
self.assertTrue(
np.allclose(tc.intensity[0, :].toarray() * 1.03375, tc_cc.intensity[0, :].toarray()))
self.assertTrue(
np.allclose(tc.intensity[2, :].toarray() * 1.03375, tc_cc.intensity[2, :].toarray()))
self.assertTrue(
np.allclose(tc.intensity[3, :].toarray() * 1.01875, tc_cc.intensity[3, :].toarray()))
res_frequency = np.ones(4) * 0.5
res_frequency[1] = 0.5 * (1 + (0.7 - 1) * scale)
res_frequency[2] = 0.5 * (1 + (2 - 1) * scale)
res_frequency[3] = 0.5 * (1 + (1.025 - 1) * scale)
self.assertTrue(np.allclose(tc_cc.frequency, res_frequency))
def test_negative_freq_error(self):
"""Test _apply_knutson_criterion with infeasible input."""
criterion = [{'basin': 'SP', 'category': [0, 1],
'year': 2100, 'change': 0.5,
'variable': 'frequency'}
]
tc = TropCyclone()
tc.frequency = np.ones(2)
tc.basin = ['SP', 'SP']
tc.category = np.array([0, 1])
with self.assertRaises(ValueError):
tc._apply_knutson_criterion(criterion, 3)
def test_two_criterion_track(self):
""" Test _apply_criterion function with two criteria """
criterion = list()
tmp_chg = {'criteria': {'basin': ['NA'], 'category':[1, 2, 3, 4, 5]},
'year': 2100, 'change': 1.045, 'variable': 'intensity', 'function': np.multiply}
criterion.append(tmp_chg)
tmp_chg = {'criteria': {'basin': ['WP'], 'category':[1, 2, 3, 4, 5]},
'year': 2100, 'change': 1.025, 'variable': 'intensity', 'function': np.multiply}
criterion.append(tmp_chg)
tmp_chg = {'criteria': {'basin': ['WP'], 'category':[1, 2, 3, 4, 5]},
'year': 2100, 'change': 1.025, 'variable': 'frequency', 'function': np.multiply}
criterion.append(tmp_chg)
scale = 0.75
tc = TropCyclone()
tc.intensity = sparse.lil_matrix(np.zeros((4, 10)))
tc.intensity[0, :] = np.arange(10)
tc.intensity[1, 5] = 10
tc.intensity[2, :] = np.arange(10, 20)
tc.intensity[3, 3] = 3
tc.intensity = tc.intensity.tocsr()
tc.frequency = np.ones(4)*0.5
tc.basin = ['NA'] * 4
tc.basin[3] = 'WP'
tc.category = np.array([2, 0, 4, 1])
tc.event_id = np.arange(4)
tc_cc = tc._apply_criterion(criterion, scale)
self.assertTrue(np.allclose(tc.intensity[1, :].todense(), tc_cc.intensity[1, :].todense()))
self.assertFalse(np.allclose(tc.intensity[3, :].todense(), tc_cc.intensity[3, :].todense()))
self.assertFalse(np.allclose(tc.intensity[0, :].todense(), tc_cc.intensity[0, :].todense()))
self.assertFalse(np.allclose(tc.intensity[2, :].todense(), tc_cc.intensity[2, :].todense()))
self.assertTrue(np.allclose(tc.intensity[0, :].todense()*1.03375, tc_cc.intensity[0, :].todense()))
self.assertTrue(np.allclose(tc.intensity[2, :].todense()*1.03375, tc_cc.intensity[2, :].todense()))
self.assertTrue(np.allclose(tc.intensity[3, :].todense()*1.01875, tc_cc.intensity[3, :].todense()))
res_frequency = np.ones(4)*0.5
res_frequency[3] = 0.5*1.01875
self.assertTrue(np.allclose(tc_cc.frequency, res_frequency))
