def test_basic_tree_functions(self):
"""Check so that the tree class behave as expected."""
with pytest.raises(SyntaxError):
tree = IntervalTree([[2, 'id01']], 1, 2)
with pytest.raises(SyntaxError):
self.simplest_tree.find_range([1])
with pytest.raises(ValueError):
self.simplest_tree.find_range([1, 'a'])
def setup_class(self):
"""Setup different interval trees and check if they behave correct"""
# Setup family with sick kid, sick father and healthy mother:
# This should behave such that pos 1 and 2 are covered by the interval:
empty_interval = [0, 0, None]
self.empty_tree = IntervalTree([empty_interval], 0, 0)
smallest_interval = [1, 1, 'id_01']
self.simplest_tree = IntervalTree([smallest_interval], 1, 2)
small_interval = [1, 2, 'id_02']
self.simple_tree = IntervalTree([smallest_interval, small_interval], 1,
2)
interval_3 = [10, 20, 'id_03']
interval_4 = [16, 40, 'id_04']
self.tree = IntervalTree([interval_3, interval_4], 1, 50)
interval_5 = [10, 20, 'id_05']
interval_6 = [10, 20, 'id_06']
interval_7 = [10, 20, 'id_06']
self.ovelapping_tree = IntervalTree(
[interval_5, interval_6, interval_7], 1, 50)
interval_8 = (10, 20, 'id_08')
self.tuple_tree = IntervalTree([interval_8], 1, 50)