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]))