/
rnekrasov
/
knowledge-graph
Обзор
Документация
Войти
/
rnekrasov
/
knowledge-graph
Код
Запросы
0
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
main
knowledge_graph/transform_test.py
32 строки
1 KB
itayzit
add tests and fix typing
08 апр 2022, 02:15
08 апр 2022, 02:15
0b07728
Код
Авторство
О чём код?
import timeit import gamla # noqa from knowledge_graph import common_relations, transform, triplets_index APPLE = "apple" ORANGE = "orange" FRUIT = "fruit" def test_merge_graphs_nodes_by_id(): kg1 = triplets_index.from_triplet(common_relations.type_triplet(APPLE, FRUIT)) kg2 = triplets_index.from_triplet(common_relations.type_triplet(ORANGE, FRUIT)) assert transform.merge_graphs_nodes_by_id( [kg1, kg2] ) == triplets_index.from_triplets( [ common_relations.type_triplet(APPLE, FRUIT), common_relations.type_triplet(ORANGE, FRUIT), ] ) # Make sure merge implementation is better than just concatenating the triplets. assert timeit.timeit( "transform.merge_graphs_nodes_by_id", setup="graphs = [kg1, kg2]", globals=globals() | locals(), ) < timeit.timeit( "lambda graphs: gamla.pipe(graphs, gamla.mapcat(triplets_index.triplets), triplets_index.from_triplets)", setup="graphs = [kg1, kg2]", globals=globals() | locals(), )