Source Code for Module visgraph.unittests.graphdbtest

 1  import visgraph.dbcore as vg_dbcore 
 2   
 3  dbinfo = { 
 4  'user':'visgraph', 
 5  'password':'ohhai!', 
 6  'database':'vg_test', 
 7  } 
 8   
 9 -def vgtest_basic_database(): 
10  #vg_dbcore.initGraphDb(dbinfo) 
11   
12      gstore = vg_dbcore.DbGraphStore(dbinfo) 
13   
14      n1 = gstore.addNode(ninfo={'name':'foo', 'size':20}) 
15      n2 = gstore.addNode(ninfo={'name':'bar', 'size':300}) 
16      n3 = gstore.addNode(ninfo={'name':'baz'}) 
17      n4 = gstore.addNode(ninfo={'name':'faz'}) 
18      n5 = gstore.addNode(ninfo={'name':'yer'}) 
19      n6 = gstore.addNode(ninfo={'name':'mom'}) 
20   
21      gstore.addEdge(n3, n4) 
22      gstore.addEdge(n4, n5) 
23      gstore.addEdge(n5, n6) 
24   
25      print gstore.getNodeInfo(n1, 'name') 
26      print gstore.getNodeInfo(n1, 'size') 
27      print gstore.getNodeInfo(n1, 'owoot', 20) 
28   
29      eid = gstore.addEdge(n1, n2, einfo={'etype':'FooEdge'}) 
30      print eid 
31      gstore.setEdgeInfo(eid, 'name', 'asdf') 
32      gstore.setEdgeInfo(eid, 'size', 20) 
33      print gstore.getEdgeInfo(eid, 'size') 
34   
35      sg = gstore.buildSubGraph() 
36   
37      sg.useEdges(size=20) 
38      #n3 = sg.addNode(ninfo={'name':'Tom Jones'}) 
39      #sg.addEdge(n2, n3, einfo={'etype':'FBFriend'}) 
40   
41      #print sg.getRefsFrom(n2) 
42   
43      for eid, fromid, toid, einfo in sg.getRefsFrom(n2): 
44          print 'NAMES: %s -> %s' % (sg.getNodeInfo(fromid, 'name', 'unknown'), sg.getNodeInfo(toid, 'name', 'unknown')) 
45   
46      sg.expandNode(n3, maxdepth=1) 
47