def test_select_wrong_pass(self):
"""Test select wrong time range."""
disc_rate = DiscRates()
disc_rate.tag.file_name = 'file1.txt'
disc_rate.tag.description = 'descr1'
disc_rate.years = np.arange(2000, 2050)
disc_rate.rates = np.arange(disc_rate.years.size)
year_range = np.arange(2050, 2060)
self.assertEqual(None, disc_rate.select(year_range))
def test_select_pass(self):
"""Test select right time range."""
disc_rate = DiscRates()
disc_rate.tag.file_name = 'file1.txt'
disc_rate.tag.description = 'descr1'
disc_rate.years = np.arange(2000, 2050)
disc_rate.rates = np.arange(disc_rate.years.size)
year_range = np.arange(2010, 2020)
sel_disc = disc_rate.select(year_range)
self.assertTrue(np.array_equal(sel_disc.years, year_range))
self.assertTrue(np.array_equal(sel_disc.rates, disc_rate.rates[10:20]))
