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