diff --git a/tests/test_nontree_vit_haploid.py b/tests/test_nontree_vit_haploid.py index d681a0c..3560c78 100644 --- a/tests/test_nontree_vit_haploid.py +++ b/tests/test_nontree_vit_haploid.py @@ -152,7 +152,16 @@ def verify(self, ts, scale_mutation_rate, include_ancestors): emission_func=emission_func, ) path_tmp = vh.backwards_viterbi_hap(m=m, V_last=V_tmp, P=P_tmp) - ll_check = vh.path_ll_hap(n=n, m=m, H=H_vs, path=path_tmp, s=s, e=e_vs, r=r, emission_func=emission_func,) + ll_check = vh.path_ll_hap( + n=n, + m=m, + H=H_vs, + path=path_tmp, + s=s, + e=e_vs, + r=r, + emission_func=emission_func, + ) self.assertAllClose(ll_tmp, ll_check) self.assertAllClose(ll_vs, ll_tmp)