def test_db(self): data = ('testKey', 'testVal') cogdb = Cog("/tmp/" + DIR_NAME + "/test") cogdb.create_namespace("test") cogdb.create_table("db_test", "test") cogdb.put(data) self.assertEqual(cogdb.get("testKey"), ('0', ('testKey', 'testVal')))
class Loader: def __init__(self, db_path=None, config=cfg): self.cog = Cog(db_path=db_path, config=config) def load_triples(self, graph_data_path, graph_name): self.cog.create_namespace(graph_name) with open(graph_data_path) as f: for line in f: tokens = line.split() this_vertex = tokens[0].strip() predicate = tokens[1].strip() other_vertex = tokens[2].strip() self.cog.create_table(predicate, graph_name) #it wont create if it exists. put_node(self.cog,this_vertex, predicate, other_vertex) def load_edgelist(self, edgelist_file_path, graph_name, predicate="none"): self.cog.create_namespace(graph_name) with open(edgelist_file_path) as f: for line in f: tokens = line.split() v1 = tokens[0].strip() v2 = tokens[1].strip() self.cog.create_table(predicate, graph_name) put_node(self.cog, v1, predicate, v2)
def test_db(self): db_path = '/tmp/cogtestdb2' try: os.makedirs(db_path) except OSError: if not os.path.isdir(db_path): raise config.CUSTOM_COG_DB_PATH = db_path cogdb = Cog() # create a namespace cogdb.create_namespace("my_namespace") # create new table cogdb.create_table("new_db", "my_namespace") # put some data cogdb.put(Record('A', 'val')) cogdb.put(Record('B', 'val')) cogdb.put(Record('key3', 'val')) cogdb.put(Record('key3', 'val_updated')) self.assertEqual(cogdb.get('key3').value, 'val_updated') cogdb.close()
def test_db(self): data = Record('testKey','testVal') cogdb = Cog() cogdb.create_namespace("test") cogdb.create_table("db_test", "test") cogdb.put(data) self.assertTrue(cogdb.get("testKey").is_equal_val(Record('testKey', 'testVal'))) cogdb.close()
def test_list_tables(self): cogdb = Cog() cogdb.create_namespace("test_ns") cogdb.create_table("table1", "test_ns") cogdb.create_table("table2", "test_ns") cogdb.create_table("table3", "test_ns") self.assertEqual(set(cogdb.list_tables()), {'table2', 'table3', 'table1'}) cogdb.close()
def test_db(self): data = ('user100', '{"firstname":"Hari","lastname":"seldon"}') cogdb = Cog(config=config) cogdb.create_namespace("test") cogdb.create_table("db_test", "test") cogdb.put(data) scanner = cogdb.scanner() for r in scanner: print r
def test_db(self): data = Record('user100','{"firstname":"Hari","lastname":"seldon"}') cogdb = Cog() cogdb.create_namespace("test") cogdb.create_table("db_test", "test") cogdb.put(data) scanner = cogdb.scanner() res = None for r in scanner: res = r print(res) self.assertTrue(data.is_equal_val(res)) cogdb.close()