class TestGraph(unittest.TestCase):
# each test_foo() method is wrapped w/ setup/teardown around it, so each test has a fresh graph
def setUp(self):
fd, path = tempfile.mkstemp()
os.close(fd)
self.g = Graph(path)
def tearDown(self):
self.g.delete()
def test_commit(self):
with self.g.transaction(write=True) as txn:
self.assertEqual(txn.nextID, 1)
n = txn.node(type='foo', value='bar')
self.assertEqual(txn.nextID, 2)
txn.commit()
# should never reach this
self.assertTrue(False)
with self.g.transaction(write=False) as txn:
self.assertEqual(txn.nextID, 2)
def test_abort(self):
with self.g.transaction(write=True) as txn:
self.assertEqual(txn.nextID, 1)
n = txn.node(type='foo', value='bar')
self.assertEqual(txn.nextID, 2)
txn.abort()
# should never reach this
self.assertTrue(False)
with self.g.transaction(write=False) as txn:
self.assertEqual(txn.nextID, 1)
def test_load_data(self):
with self.g.transaction(write=True) as txn:
load_data(txn)
def test_edges_by_type(self):
with self.g.transaction(write=True) as txn:
load_data(txn)
n1 = node(1)
n1 = txn.node(**n1)
all_edges = sum(1 for x in n1.iterlinks())
fewer_edges = sum(1 for x in n1.iterlinks(types=(edge(1)['type'],)))
self.assertTrue(all_edges > fewer_edges)
def test_edge_dirs(self):
with self.g.transaction(write=True) as txn:
load_data(txn)
with self.g.transaction(write=False) as txn:
n1 = node(1)
n1 = txn.node(**n1)
all_edges = set(e.ID for e in n1.edges)
out_edges = set(e.ID for e in n1.edges(dir="out"))
in_edges = set(e.ID for e in n1.edges(dir="in"))
self.assertTrue(all_edges)
self.assertTrue(out_edges)
self.assertTrue(in_edges)
self.assertTrue(all_edges - in_edges)
self.assertTrue(all_edges - out_edges)
self.assertTrue(in_edges.isdisjoint(out_edges))
def test_query(self):
with self.g.transaction(write=True) as txn:
load_data(txn)
with self.g.transaction(write=False) as txn:
chains = 0
for chain in txn.query("n(type='foo')->e()-n()"):
chains += 1
self.assertTrue(chains)
def test_graph_props(self):
with self.g.transaction(write=True) as txn:
self.assertFalse('foo' in txn)
txn['foo'] = 'bar'
self.assertTrue('foo' in txn)
def test_kv(self):
with self.g.transaction(write=True) as txn:
b = txn.kv('foo')
keys =('fon', 'foo', 'foobar', 'foobaz', 'fom')
for i, key in enumerate(('fo', 'fon', 'foo', 'foobar', 'foobaz', 'fom')):
b[key] = i
res = tuple(b.iterkeys(pfx='foo'))
self.assertEqual( keys[1:-1], res )
txn.abort()
with self.g.transaction(write=True) as txn:
b = txn.kv('foo', map_data=True)
self.assertFalse('bar' in b)
b['bar'] = 'blah'
self.assertTrue('bar' in b)
txn.abort()
with self.g.transaction(write=True) as txn:
b = txn.kv('foo', map_keys=True)
self.assertFalse('bar' in b)
b['bar'] = 'blah'
self.assertTrue('bar' in b)
count = sum(1 for x in b)
self.assertTrue(1 == count)
def test_fifo(self):
with self.g.transaction(write=True) as txn:
f = txn.fifo('foo', serialize_value=Serializer.uint())
f.push(1, 2, 3)
out = f.pop(4)
self.assertEqual(out, (1, 2, 3))
self.assertTrue(f.empty)
f.push(4, 5, 6)
self.assertFalse(f.empty)
self.assertEqual(len(f), 3)
out = f.pop(4)
self.assertEqual(out, (4, 5, 6))
self.assertEqual(len(f), 0)
def test_nested(self):
with self.g.transaction(write=True) as t0:
t0['foo'] = "t0"
with t0.transaction(write=True) as t1:
t1['foo'] = "t1"
with t1.transaction(write=True) as t2:
self.assertTrue(t2['foo'] == "t1")
t2['foo'] = "t2"
self.assertTrue(t2['foo'] == "t2")
t1.abort()
# should never get here
self.assertTrue(False)
self.assertTrue(t0['foo'] == "t0")
t0['foo'] = "t00"
with self.g.transaction(write=True) as t0:
self.assertTrue(t0['foo'] == "t00")
t0['foo'] = "t000"
with t0.transaction(write=True) as t1:
t1['foo'] = "t1"
t0.commit()
# should never get here
self.assertTrue(False)
# or here
self.assertTrue(False)
with self.g.transaction(write=True) as t0:
self.assertTrue(t0['foo'] == "t000")
def test_reset(self):
with self.g.transaction(write=True) as txn:
n = txn.node(type="foo", value="bar")
nextID = txn.nextID
with self.g.transaction(write=True) as txn:
self.assertEqual(nextID, txn.nextID)
txn.reset()
self.assertEqual(1, txn.nextID)
with self.g.transaction(write=False) as txn:
self.assertEqual(1, txn.nextID)
class TestAlgorithms(unittest.TestCase):
serializer = Serializer.msgpack()
# each test_foo() method is wrapped w/ setup/teardown around it, so each test has a fresh graph
def setUp(self):
fd, path = tempfile.mkstemp()
os.close(fd)
self.g = Graph(path, serialize_property_value=self.serializer)
def tearDown(self):
self.g.delete()
def test_sp1(self):
with self.g.transaction(write=True) as txn:
load_data(txn)
with self.g.transaction(write=False) as txn:
n0 = txn.node(**node(0))
n1 = txn.node(**node(1))
n2 = txn.node(**node(2))
e0 = edge(0)
e1 = edge(1)
e0['src'] = txn.node(**e0['src'])
e0['tgt'] = txn.node(**e0['tgt'])
e1['src'] = txn.node(**e1['src'])
e1['tgt'] = txn.node(**e1['tgt'])
e0 = txn.edge(**e0)
e1 = txn.edge(**e1)
expect_path = (n0, e0, n1, e1, n2)
res_path = n0.shortest_path(n2, directed=False)
self.assertEqual(
tuple(x.ID for x in expect_path),
tuple(x.ID for x in res_path))
res_path = n0.shortest_path(n2, directed=True)
self.assertEqual(
tuple(x.ID for x in expect_path),
tuple(x.ID for x in res_path))
fail = n2.shortest_path(n0, directed=True)
self.assertEqual(fail, None)
def test_sp2(self):
with self.g.transaction(write=True) as txn:
'''
n1a
/ \
e0a e1a
/ \
n0 n2
\ /
e0b e1b
\ /
n1b
'''
n0 = txn.node(type='foo', value='0')
n1a = txn.node(type='foo', value='1a')
n1b = txn.node(type='foo', value='1b')
n2 = txn.node(type='foo', value='2')
e0a = txn.edge(type='foo', src=n0, tgt=n1a)
e0b = txn.edge(type='foo', src=n0, tgt=n1b)
e1a = txn.edge(type='foo', src=n1a, tgt=n2)
e1b = txn.edge(type='foo', src=n1b, tgt=n2)
# should transit upper path
self.assertPathEqual(n0.shortest_path(n2), (n0, e0a, n1a, e1a, n2))
# use cost to force it through lower path
e1b['cost'] = 0.5
self.assertPathEqual(n0.shortest_path(n2, cost_field='cost'), (n0, e0b, n1b, e1b, n2))
# use default cost to make it find the upper path again
self.assertPathEqual(n0.shortest_path(n2, cost_field='cost', cost_default=0.5), (n0, e0a, n1a, e1a, n2))
def assertPathEqual(self, a, b):
self.assertEqual(type(a), type(b))
if a is not None:
self.assertEqual(
tuple(x.ID for x in a),
tuple(x.ID for x in b))
class TestGraph(unittest.TestCase):
# each test_foo() method is wrapped w/ setup/teardown around it, so each test has a fresh graph
def setUp(self):
fd, path = tempfile.mkstemp()
os.close(fd)
self.g = Graph(path)
def tearDown(self):
self.g.delete()
def test_commit(self):
with self.g.transaction(write=True) as txn:
self.assertEqual(txn.nextID, 1)
txn.node(type='foo', value='bar')
self.assertEqual(txn.nextID, 2)
txn.commit()
# should never reach this
self.assertTrue(False)
with self.g.transaction(write=False) as txn:
self.assertEqual(txn.nextID, 2)
def test_abort(self):
with self.g.transaction(write=True) as txn:
self.assertEqual(txn.nextID, 1)
txn.node(type='foo', value='bar')
self.assertEqual(txn.nextID, 2)
txn.abort()
# should never reach this
self.assertTrue(False)
with self.g.transaction(write=False) as txn:
self.assertEqual(txn.nextID, 1)
def test_load_data(self):
with self.g.transaction(write=True) as txn:
load_data(txn)
self.assertEqual(txn.nodes_count(), len(Nodes))
self.assertEqual(txn.edges_count(), len(Edges))
def test_counts(self):
with self.g.transaction(write=True) as txn:
self.assertEqual(txn.nodes_count(), 0)
self.assertEqual(txn.edges_count(), 0)
n1 = txn.node(type="foo", value="bar")
self.assertEqual(txn.nodes_count(), 1)
n2 = txn.node(type="foo", value="baz")
self.assertEqual(txn.nodes_count(), 2)
self.assertEqual(txn.nodes_count(beforeID=n2.ID), 1)
txn.edge(src=n1, tgt=n2, type="foo")
self.assertEqual(txn.edges_count(), 1)
with self.g.transaction(write=True) as txn:
self.assertEqual(txn.nodes_count(beforeID=n2.ID), 1)
txn.node(type="foo", value="blah")
self.assertEqual(txn.nodes_count(), 3)
n1 = txn.node(type="foo", value="bar")
n1.delete()
self.assertEqual(txn.nodes_count(), 2)
self.assertEqual(txn.edges_count(), 0)
with self.g.transaction(write=False) as txn:
self.assertEqual(txn.nodes_count(), 2)
self.assertEqual(txn.edges_count(), 0)
self.assertEqual(txn.nodes_count(beforeID=txn.lastID), 3)
def test_edges_by_type(self):
with self.g.transaction(write=True) as txn:
load_data(txn)
n1 = node(1)
n1 = txn.node(**n1)
all_edges = sum(1 for x in n1.iterlinks())
fewer_edges = sum(1 for x in n1.iterlinks(types=(edge(1)['type'],)))
self.assertTrue(all_edges > fewer_edges)
def test_edge_dirs(self):
with self.g.transaction(write=True) as txn:
load_data(txn)
with self.g.transaction(write=False) as txn:
n1 = node(1)
n1 = txn.node(**n1)
all_edges = set(e.ID for e in n1.edges)
out_edges = set(e.ID for e in n1.edges(dir="out"))
in_edges = set(e.ID for e in n1.edges(dir="in"))
self.assertTrue(all_edges)
self.assertTrue(out_edges)
self.assertTrue(in_edges)
self.assertTrue(all_edges - in_edges)
self.assertTrue(all_edges - out_edges)
self.assertTrue(in_edges.isdisjoint(out_edges))
def test_query(self):
with self.g.transaction(write=True) as txn:
load_data(txn)
with self.g.transaction(write=False) as txn:
chains = 0
for _ in txn.query("n(type='foo')->e()-n()"):
chains += 1
self.assertTrue(chains)
def test_graph_props(self):
with self.g.transaction(write=True) as txn:
self.assertFalse('foo' in txn)
txn['foo'] = 'bar'
self.assertTrue('foo' in txn)
def test_kv(self):
with self.g.transaction(write=True) as txn:
b = txn.kv('foo')
keys = ('fon', 'foo', 'foobar', 'foobaz', 'fom')
for i, key in enumerate(('fo', 'fon', 'foo', 'foobar', 'foobaz', 'fom')):
b[key] = i
res = tuple(b.iterkeys(pfx='foo'))
self.assertEqual(keys[1:-1], res)
txn.abort()
with self.g.transaction(write=True) as txn:
b = txn.kv('foo', map_data=True)
self.assertFalse('bar' in b)
b['bar'] = 'blah'
self.assertTrue('bar' in b)
txn.abort()
with self.g.transaction(write=True) as txn:
b = txn.kv('foo', map_keys=True)
self.assertFalse('bar' in b)
b['bar'] = 'blah'
self.assertTrue('bar' in b)
count = sum(1 for x in b)
self.assertTrue(1 == count)
def test_fifo(self):
with self.g.transaction(write=True) as txn:
f = txn.fifo('foo', serialize_value=Serializer.uint())
f.push(1, 2, 3)
out = f.pop(4)
self.assertEqual(out, (1, 2, 3))
self.assertTrue(f.empty)
f.push(4, 5, 6)
self.assertFalse(f.empty)
self.assertEqual(len(f), 3)
out = f.pop(4)
self.assertEqual(out, (4, 5, 6))
self.assertEqual(len(f), 0)
def test_nested(self):
with self.g.transaction(write=True) as t0:
t0['foo'] = "t0"
with t0.transaction(write=True) as t1:
t1['foo'] = "t1"
with t1.transaction(write=True) as t2:
self.assertTrue(t2['foo'] == "t1")
t2['foo'] = "t2"
self.assertTrue(t2['foo'] == "t2")
t1.abort()
# should never get here
self.assertTrue(False)
self.assertTrue(t0['foo'] == "t0")
t0['foo'] = "t00"
with self.g.transaction(write=True) as t0:
self.assertTrue(t0['foo'] == "t00")
t0['foo'] = "t000"
with t0.transaction(write=True) as t1:
t1['foo'] = "t1"
t0.commit()
# should never get here
self.assertTrue(False)
# or here
self.assertTrue(False)
with self.g.transaction(write=True) as t0:
self.assertTrue(t0['foo'] == "t000")
def test_reset(self):
with self.g.transaction(write=True) as txn:
txn.node(type="foo", value="bar")
nextID = txn.nextID
with self.g.transaction(write=True) as txn:
self.assertEqual(nextID, txn.nextID)
txn.reset()
self.assertEqual(1, txn.nextID)
with self.g.transaction(write=False) as txn:
self.assertEqual(1, txn.nextID)
class TestAlgorithms(unittest.TestCase):
serializer = Serializer.msgpack()
# each test_foo() method is wrapped w/ setup/teardown around it, so each test has a fresh graph
def setUp(self):
fd, path = tempfile.mkstemp()
os.close(fd)
self.g = Graph(path, serialize_property_value=self.serializer)
def tearDown(self):
self.g.delete()
def test_sp1(self):
with self.g.transaction(write=True) as txn:
load_data(txn)
with self.g.transaction(write=False) as txn:
n0 = txn.node(**node(0))
n1 = txn.node(**node(1))
n2 = txn.node(**node(2))
e0 = edge(0)
e1 = edge(1)
e0['src'] = txn.node(**e0['src'])
e0['tgt'] = txn.node(**e0['tgt'])
e1['src'] = txn.node(**e1['src'])
e1['tgt'] = txn.node(**e1['tgt'])
e0 = txn.edge(**e0)
e1 = txn.edge(**e1)
expect_path = (n0, e0, n1, e1, n2)
res_path = n0.shortest_path(n2, directed=False)
self.assertEqual(
tuple(x.ID for x in expect_path),
tuple(x.ID for x in res_path))
res_path = n0.shortest_path(n2, directed=True)
self.assertEqual(
tuple(x.ID for x in expect_path),
tuple(x.ID for x in res_path))
fail = n2.shortest_path(n0, directed=True)
self.assertEqual(fail, None)
def test_sp2(self):
with self.g.transaction(write=True) as txn:
'''
n1a
/ \
e0a e1a
/ \
n0 n2
\ /
e0b e1b
\ /
n1b
'''
n0 = txn.node(type='foo', value='0')
n1a = txn.node(type='foo', value='1a')
n1b = txn.node(type='foo', value='1b')
n2 = txn.node(type='foo', value='2')
e0a = txn.edge(type='foo', src=n0, tgt=n1a)
e0b = txn.edge(type='foo', src=n0, tgt=n1b)
e1a = txn.edge(type='foo', src=n1a, tgt=n2)
e1b = txn.edge(type='foo', src=n1b, tgt=n2)
# should transit upper path
self.assertPathEqual(n0.shortest_path(n2), (n0, e0a, n1a, e1a, n2))
# use cost to force it through lower path
e1b['cost'] = 0.5
self.assertPathEqual(n0.shortest_path(n2, cost_field='cost'), (n0, e0b, n1b, e1b, n2))
# use default cost to make it find the upper path again
self.assertPathEqual(n0.shortest_path(n2, cost_field='cost', cost_default=0.5), (n0, e0a, n1a, e1a, n2))
def assertPathEqual(self, a, b):
self.assertEqual(type(a), type(b))
if a is not None:
self.assertEqual(
tuple(x.ID for x in a),
tuple(x.ID for x in b))
