File tree 1 file changed +13
-6
lines changed
1 file changed +13
-6
lines changed Original file line number Diff line number Diff line change @@ -286,9 +286,16 @@ def test_toy_example():
286
286
print (list (ref_ts .haplotypes ()))
287
287
print (ref_ts )
288
288
print (query )
289
- fw , bw = get_tskit_forward_backward_matrices (ref_ts , query [0 ])
290
- print (fw )
291
- print (bw )
292
- df_beagle = parse_matrix (beagle_fwd_matrix_text_1 )
293
- # compare beagle results with tskit
294
- print (df_beagle ["val" ])
289
+ tskit_fwd , tskit_bwd = get_tskit_forward_backward_matrices (ref_ts , query [0 ])
290
+ beagle_fwd = parse_matrix (beagle_fwd_matrix_text_1 )
291
+ beagle_bwd = parse_matrix (beagle_bwd_matrix_text_1 )
292
+ print ("Forward probability matrix" )
293
+ print ("tskit" )
294
+ print (tskit_fwd )
295
+ print ("beagle" )
296
+ print (beagle_fwd ["val" ].reshape ((4 , 6 )))
297
+ print ("Backward probability matrix" )
298
+ print ("tskit" )
299
+ print (tskit_bwd )
300
+ print ("beagle" )
301
+ print (beagle_bwd ["val" ].reshape ((4 , 6 )))
You can’t perform that action at this time.
0 commit comments