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