Skip to content

Commit 845ccec

Browse files
szhanmergify[bot]
authored andcommitted
Fix bugs
1 parent 22fb89a commit 845ccec

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

python/tests/test_highlevel.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3286,13 +3286,15 @@ def test_empty_files(self):
32863286
edges_file = io.StringIO("left\tright\tparent\tchild\n")
32873287
sites_file = io.StringIO("position\tancestral_state\n")
32883288
mutations_file = io.StringIO("site\tnode\tderived_state\n")
3289+
individuals_file = io.StringIO("flags\n")
32893290
migrations_file = io.StringIO("left\tright\tnode\tsource\tdest\ttime\n")
32903291
with pytest.raises(_tskit.LibraryError):
32913292
tskit.load_text(
32923293
nodes=nodes_file,
32933294
edges=edges_file,
32943295
sites=sites_file,
32953296
mutations=mutations_file,
3297+
individuals=individuals_file,
32963298
migrations=migrations_file,
32973299
)
32983300

@@ -3301,20 +3303,23 @@ def test_empty_files_sequence_length(self):
33013303
edges_file = io.StringIO("left\tright\tparent\tchild\n")
33023304
sites_file = io.StringIO("position\tancestral_state\n")
33033305
mutations_file = io.StringIO("site\tnode\tderived_state\n")
3306+
individuals_file = io.StringIO("flags\n")
33043307
migrations_file = io.StringIO("left\tright\tnode\tsource\tdest\ttime\n")
33053308
ts = tskit.load_text(
33063309
nodes=nodes_file,
33073310
edges=edges_file,
33083311
sites=sites_file,
33093312
mutations=mutations_file,
3313+
individuals=individuals_file,
33103314
migrations=migrations_file,
33113315
sequence_length=100,
33123316
)
33133317
assert ts.sequence_length == 100
33143318
assert ts.num_nodes == 0
33153319
assert ts.num_edges == 0
33163320
assert ts.num_sites == 0
3317-
assert ts.num_edges == 0
3321+
assert ts.num_mutations == 0
3322+
assert ts.num_individuals == 0
33183323
assert ts.num_migrations == 0
33193324

33203325
def test_load_text_no_populations(self):

0 commit comments

Comments
 (0)