def test_append_different_append(self): """Append DiscRates with same and new values. The rates with repeated years are overwritten.""" disc_rate = DiscRates() disc_rate.tag.file_name = 'file1.txt' disc_rate.tag.description = 'descr1' disc_rate.years = np.array([2000, 2001, 2002]) disc_rate.rates = np.array([0.1, 0.2, 0.3]) disc_rate_add = DiscRates() disc_rate_add.tag.file_name = 'file2.txt' disc_rate_add.tag.description = 'descr2' disc_rate_add.years = np.array([2000, 2001, 2003]) disc_rate_add.rates = np.array([0.11, 0.22, 0.33]) disc_rate.append(disc_rate_add) disc_rate.check() self.assertTrue( np.array_equal(disc_rate.years, np.array([2000, 2001, 2002, 2003]))) self.assertTrue( np.array_equal(disc_rate.rates, np.array([0.11, 0.22, 0.3, 0.33]))) self.assertTrue( np.array_equal(disc_rate.tag.file_name, 'file1.txt + file2.txt')) self.assertTrue( np.array_equal(disc_rate.tag.description, 'descr1 + descr2'))
def test_append_to_empty_same(self): """Append DiscRates to empty one.""" disc_rate = DiscRates() disc_rate_add = DiscRates() disc_rate_add.tag.file_name = 'file1.txt' disc_rate_add.tag.description = 'descr1' disc_rate_add.years = np.array([2000, 2001, 2002]) disc_rate_add.rates = np.array([0.1, 0.2, 0.3]) disc_rate.append(disc_rate_add) disc_rate.check() self.assertTrue(np.array_equal(disc_rate.years, disc_rate_add.years)) self.assertTrue(np.array_equal(disc_rate.rates, disc_rate_add.rates)) self.assertTrue(np.array_equal(disc_rate.tag.file_name, \ disc_rate_add.tag.file_name)) self.assertTrue(np.array_equal(disc_rate.tag.description, \ disc_rate_add.tag.description))
def test_append_equal_same(self): """Append the same DiscRates. The inital DiscRates is obtained.""" disc_rate = DiscRates() disc_rate.tag.file_name = 'file1.txt' disc_rate.tag.description = 'descr1' disc_rate.years = np.array([2000, 2001, 2002]) disc_rate.rates = np.array([0.1, 0.2, 0.3]) disc_rate_add = DiscRates() disc_rate_add.tag.file_name = 'file1.txt' disc_rate_add.tag.description = 'descr1' disc_rate_add.years = np.array([2000, 2001, 2002]) disc_rate_add.rates = np.array([0.1, 0.2, 0.3]) disc_rate.append(disc_rate_add) disc_rate.check() self.assertTrue(np.array_equal(disc_rate.years, disc_rate_add.years)) self.assertTrue(np.array_equal(disc_rate.rates, disc_rate_add.rates)) self.assertTrue(np.array_equal(disc_rate.tag.file_name, disc_rate_add.tag.file_name)) self.assertEqual(disc_rate.tag.description, disc_rate_add.tag.description)