Skip to content

Commit f20d91b

Browse files
committed
Add function to populate databse
1 parent b1f13b6 commit f20d91b

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

source_code/intersection_trees/intersection_tree.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,3 +223,13 @@ def create_queries(size: int = 1_000, max_end: int = 1_000_000) -> Queries:
223223
a list of half-open intervals
224224
'''
225225
return [generate_interval(max_end) for _ in range(size)]
226+
227+
228+
def populate_db(db: Node | None, intervals: typing.Sequence[Interval]) -> Node:
229+
if len(intervals) == 0:
230+
raise ValueError('At least 1 interval is required')
231+
if db is None:
232+
db = Node(intervals.pop(0))
233+
for interval in intervals:
234+
db.insert(interval)
235+
return db

0 commit comments

Comments
 (0)