DEV: generalise the build_tree tests to rooted topologies

iqtree · Aug 22, 2024 · 79de514 · 79de514
1 parent 0045a02
commit 79de514
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_iqtree/test_build_tree.py b/tests/test_iqtree/test_build_tree.py
@@ -41,8 +41,10 @@ def test_build_tree(four_otu, dna_model, freq_type, invariable_sites, rate_model
     )
 
     got1 = piqtree2.build_tree(four_otu, model, rand_seed=1)
-    assert expected.same_topology(got1)
+    got1 = got1.unrooted()
+    assert expected.same_topology(got1.unrooted())
 
     # Should be similar for any seed
     got2 = piqtree2.build_tree(four_otu, model, rand_seed=None)
+    got2 = got2.unrooted()
     assert expected.same_topology(got2)