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 setUp(self): # new connection for just this test self.clean_connection = cache.cache_connect(':memory:')