def test_add_health_to_gene_trie_when_add_to_beginning():
trie = Trie('a')
trie.value = [DataItem(2, 8)]
index = 1
health = 4
add_health_to_gene_trie(trie, health, index)
result = trie.value
assert len(result) == 2
assert result[0].index == 1
assert result[0].sum == 4
assert result[1].index == 2
assert result[1].sum == 12
def test_add_health_to_gene_trie_when_add_to_end():
trie = Trie('a')
trie.value = [DataItem(2, 8), DataItem(5, 14)]
index = 7
health = 4
add_health_to_gene_trie(trie, health, index)
result = trie.value
assert len(result) == 3
assert result[0].index == 2
assert result[0].sum == 8
assert result[1].index == 5
assert result[1].sum == 14
assert result[2].index == 7
assert result[2].sum == 18
def test_add_health_to_gene_trie_when_inserting_in_middle_should_insert_in_correct_position_and_update_forward_sum():
trie = Trie('a')
trie.value = [DataItem(2, 8), DataItem(5, 14), DataItem(6, 19)]
index = 3
health = 4
add_health_to_gene_trie(trie, health, index)
result = trie.value
assert len(result) == 4
assert result[0].index == 2
assert result[0].sum == 8
assert result[1].index == 3
assert result[1].sum == 12
assert result[2].index == 5
assert result[2].sum == 18
assert result[3].index == 6
assert result[3].sum == 23