def test_viterbi_ref_1(): """ test_viterbi_ref_1 """ traj = simple_traj1() theta = np.array([1.0, 0.0]) viterbi = TrajectoryViterbiRef(traj, theta) viterbi.computeMostLikely() assert len(viterbi.most_likely) == traj.L assert viterbi.most_likely[0] == 0 assert viterbi.most_likely[1] == 0
def test_viterbi_1_7(): """ test_viterbi_1_7 """ traj = simple_traj7() theta = np.array([1.0]) viterbi_ref = TrajectoryViterbiRef(traj, theta) viterbi_ref.computeMostLikely() viterbi_1 = TrajectoryViterbi1(traj, theta) viterbi_1.computeMostLikely() assert len(viterbi_1.most_likely) == traj.L for l in range(traj.L): assert viterbi_1.most_likely[l] == viterbi_ref.most_likely[l] assert traj.num_choices[l] == len(viterbi_ref.most_likely_tree[l]) for i in range(traj.num_choices[l]): assert viterbi_ref.most_likely_tree[l][i] == \ viterbi_1.most_likely_tree[l][i]