def test_old_carry_over(self): A_prev = ['aaa', 'aab', 'aac'] l_last = np.asarray([0, 1, 2]) best_inds = (np.asarray([0, 1, 2]), np.asarray([3, 3, 3])) A_new, l_last_new = find_new_prefixes(l_last, best_inds, A_prev, self.letters, self.blank_ind) self.assertEqual(A_new, A_prev) self.assertEqual(set(l_last_new.tolist()), set(l_last.tolist()))
def test_regression1(self): A_prev = ['b', 'a'] l_last = np.asarray([1, 0]) best_inds = (np.asarray([1, 1]), np.asarray([3, 1])) A_exp = ['ab', 'a'] l_last_exp = np.asarray([1, 0]) A_new, l_last_new = find_new_prefixes(l_last, best_inds, A_prev, self.letters, self.blank_ind) self.assertEqual(set(A_new), set(A_exp)) self.assertEqual(set(l_last_new.tolist()), set(l_last_exp.tolist()))
def test_all_mixed(self): A_prev = ['aaa', 'aab', 'aac'] l_last = np.asarray([0, 1, 2]) best_inds = (np.asarray([0, 1, 2]), np.asarray([1, 3, 0])) A_exp = ['aaab', 'aab', 'aaca'] l_last_exp = np.asarray([1, 1, 0]) A_new, l_last_new = find_new_prefixes(l_last, best_inds, A_prev, self.letters, self.blank_ind) self.assertEqual(set(A_new), set(A_exp)) self.assertEqual(set(l_last_new.tolist()), set(l_last_exp.tolist()))