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