def test_insert_embedding(self):
test_database_path = cache.cache_file(filename=tmp_database_name)
conn = cache.cache_connect(test_database_path)
source_nodes = [0, 1, 2]
source_edges = [[0, 1], [0, 2], [1, 2]]
target_nodes = [0, 1, 2, 3]
target_edges = [[0, 1], [0, 3], [1, 2], [2, 3]]
embedding = {0: [0], 1: [1], 2: [2, 3]}
with conn as cur:
cache.insert_embedding(cur, source_nodes, source_edges, target_nodes, target_edges,
embedding, 'tag1')
embedding_ = cache.select_embedding_from_tag(cur, 'tag1', target_nodes, target_edges)
self.assertEqual(embedding, embedding_)
# now reinsert but with a different embedding
embedding = {0: [0, 1], 1: [2], 2: [3]}
with conn as cur:
cache.insert_embedding(cur, source_nodes, source_edges, target_nodes, target_edges,
embedding, 'tag1')
# get it back
embedding_ = cache.select_embedding_from_tag(cur, 'tag1', target_nodes, target_edges)
self.assertEqual(embedding, embedding_)
# get it back from source graph
embedding_ = cache.select_embedding_from_source(cur, source_nodes, source_edges,
target_nodes, target_edges)
self.assertEqual(embedding, embedding_)
def test_same_database(self):
"""multiple calls to get_database_path"""
db1 = cache.cache_file()
db2 = cache.cache_file()
self.assertEqual(db1, db2)