@@ -4625,7 +4625,7 @@ class TestSeekDirection:
4625
4625
def ts (self ):
4626
4626
return tsutil .all_trees_ts (3 )
4627
4627
4628
- def setup (self ):
4628
+ def get_tree_pair (self ):
4629
4629
ts = self .ts ()
4630
4630
t1 = tskit .Tree (ts )
4631
4631
t2 = tskit .Tree (ts )
@@ -4638,7 +4638,7 @@ def setup(self):
4638
4638
def test_index_from_different_directions (self , index ):
4639
4639
# Check that we get different orderings of the children arrays
4640
4640
# for all trees when we go in different directions.
4641
- t1 , t2 = self .setup ()
4641
+ t1 , t2 = self .get_tree_pair ()
4642
4642
while t1 .index != index :
4643
4643
t1 .next ()
4644
4644
while t2 .index != index :
@@ -4647,7 +4647,7 @@ def test_index_from_different_directions(self, index):
4647
4647
4648
4648
@pytest .mark .parametrize ("position" , [0 , 1 , 2 , 3 ])
4649
4649
def test_seek_from_null (self , position ):
4650
- t1 , t2 = self .setup ()
4650
+ t1 , t2 = self .get_tree_pair ()
4651
4651
t1 .clear ()
4652
4652
t1 .seek (position )
4653
4653
t2 .first ()
@@ -4656,7 +4656,7 @@ def test_seek_from_null(self, position):
4656
4656
4657
4657
@pytest .mark .parametrize ("index" , range (3 ))
4658
4658
def test_seek_next_tree (self , index ):
4659
- t1 , t2 = self .setup ()
4659
+ t1 , t2 = self .get_tree_pair ()
4660
4660
while t1 .index != index :
4661
4661
t1 .next ()
4662
4662
t2 .next ()
@@ -4666,7 +4666,7 @@ def test_seek_next_tree(self, index):
4666
4666
4667
4667
@pytest .mark .parametrize ("index" , [3 , 2 , 1 ])
4668
4668
def test_seek_prev_tree (self , index ):
4669
- t1 , t2 = self .setup ()
4669
+ t1 , t2 = self .get_tree_pair ()
4670
4670
while t1 .index != index :
4671
4671
t1 .prev ()
4672
4672
t2 .prev ()
@@ -4675,52 +4675,52 @@ def test_seek_prev_tree(self, index):
4675
4675
assert_trees_identical (t1 , t2 )
4676
4676
4677
4677
def test_seek_1_from_0 (self ):
4678
- t1 , t2 = self .setup ()
4678
+ t1 , t2 = self .get_tree_pair ()
4679
4679
t1 .first ()
4680
4680
t1 .next ()
4681
4681
t2 .first ()
4682
4682
t2 .seek (1 )
4683
4683
assert_trees_identical (t1 , t2 )
4684
4684
4685
4685
def test_seek_1_5_from_0 (self ):
4686
- t1 , t2 = self .setup ()
4686
+ t1 , t2 = self .get_tree_pair ()
4687
4687
t1 .first ()
4688
4688
t1 .next ()
4689
4689
t2 .first ()
4690
4690
t2 .seek (1.5 )
4691
4691
assert_trees_identical (t1 , t2 )
4692
4692
4693
4693
def test_seek_1_5_from_1 (self ):
4694
- t1 , t2 = self .setup ()
4694
+ t1 , t2 = self .get_tree_pair ()
4695
4695
for _ in range (2 ):
4696
4696
t1 .next ()
4697
4697
t2 .next ()
4698
4698
t2 .seek (1.5 )
4699
4699
assert_trees_identical (t1 , t2 )
4700
4700
4701
4701
def test_seek_3_from_null (self ):
4702
- t1 , t2 = self .setup ()
4702
+ t1 , t2 = self .get_tree_pair ()
4703
4703
t1 .last ()
4704
4704
t2 .seek (3 )
4705
4705
assert_trees_identical (t1 , t2 )
4706
4706
4707
4707
def test_seek_3_from_null_prev (self ):
4708
- t1 , t2 = self .setup ()
4708
+ t1 , t2 = self .get_tree_pair ()
4709
4709
t1 .last ()
4710
4710
t1 .prev ()
4711
4711
t2 .seek (3 )
4712
4712
t2 .prev ()
4713
4713
assert_trees_identical (t1 , t2 )
4714
4714
4715
4715
def test_seek_3_from_0 (self ):
4716
- t1 , t2 = self .setup ()
4716
+ t1 , t2 = self .get_tree_pair ()
4717
4717
t1 .last ()
4718
4718
t2 .first ()
4719
4719
t2 .seek (3 )
4720
4720
assert_trees_identical (t1 , t2 )
4721
4721
4722
4722
def test_seek_0_from_3 (self ):
4723
- t1 , t2 = self .setup ()
4723
+ t1 , t2 = self .get_tree_pair ()
4724
4724
t1 .last ()
4725
4725
t1 .first ()
4726
4726
t2 .last ()
0 commit comments