コード例 #1
0
ファイル: serializer.py プロジェクト: rv816/graphlib
    def test(self):
        s = Serializer()
        n = Node({'foo': 'bar', 'bar': 1})

        # Add properties for coverage
        n.labels = ['Special']
        n.match_props = ['foo']
        n.update_props = ['bar']

        rels = n.relate([Node() for _ in range(5)], 'NEXT')

        o = Node()
        n.relate(o, 'OTHER')

        # Add properties for coverage
        rels[0].match_props = ['foo']
        rels[0].update_props = ['bar']

        # Single relationship (and start and end node)
        s.serialize(rels[0])

        # Remaining rels
        s.serialize(rels[1:])

        s.serialize(n)

        # No effect
        s.serialize(n, traverse=False)
        s.serialize(rels[0])

        self.assertRaises(TypeError, s.serialize, None)

        self.assertEqual(len(s.items), 13)
        self.assertEqual(len(s.batches), 2)
コード例 #2
0
ファイル: neo4j.py プロジェクト: rv816/graphlib
 def test_parse_merge_rel(self):
     n = Node()
     r = n.relate(Node(), 'TO', {'foo': 1, 'bar': 'a'}, match_props=['foo'])
     s = "MERGE (x0)-[x2:TO {foo: 1}]->(x1) " \
         "ON CREATE SET x2 = {bar: 'a', foo: 1} " \
         "ON MATCH SET x2.bar = 'a', x2.foo = 1"
     # Third statement.. after creating the nodes
     self.assertEqual(neo4j.parse(serialize(r))[2], s)
コード例 #3
0
ファイル: neo4j.py プロジェクト: bruth/graphlib
 def test_parse_merge_rel(self):
     n = Node()
     r = n.relate(Node(), 'TO', {'foo': 1, 'bar': 'a'}, match_props=['foo'])
     s = "MERGE (x0)-[x2:TO {foo: 1}]->(x1) " \
         "ON CREATE SET x2 = {bar: 'a', foo: 1} " \
         "ON MATCH SET x2.bar = 'a', x2.foo = 1"
     # Third statement.. after creating the nodes
     self.assertEqual(neo4j.parse(serialize(r))[2], s)
コード例 #4
0
ファイル: graph.py プロジェクト: rv816/graphlib
    def test_sort(self):
        n0 = Node({'order': 0})
        n1 = Node({'order': 1})
        n2 = Node({'order': 2})
        items = Nodes([n1, n2, n0])

        self.assertCountEqual(items.sort('order'), [n0, n1, n2])

        f = lambda n: n.props.get('order')
        self.assertCountEqual(items.sort(f), [n0, n1, n2])
コード例 #5
0
ファイル: graph.py プロジェクト: rv816/graphlib
    def test(self):
        n0, n1, n2, n3 = Node(), Node(), Node(), Node()

        n0.relate([n1, n2, n3], 'CONTAINS')
        n0.relate([n1, n2], 'RELATED')

        # Get all relationships from n0
        rels = n0.rels()
        nodes = rels.nodes()

        # Gets distinct set of end nodes of rels
        self.assertCountEqual(nodes, [n1, n2, n3])
コード例 #6
0
ファイル: neo4j.py プロジェクト: bruth/graphlib
    def test_parse_create_rel(self):
        r = Node().relate(Node(), 'TO')
        s = 'MERGE (x0)-[x2:TO]->(x1)'
        # Third statement.. after creating the nodes
        self.assertEqual(neo4j.parse(serialize(r))[2], s)

        r['foo'] = 1
        r['bar'] = 'a'
        r.match_props = False
        s = "CREATE (x0)-[:TO {bar: 'a', foo: 1}]->(x1)"
        # Third statement.. after creating the nodes
        self.assertEqual(neo4j.parse(serialize(r))[2], s)
コード例 #7
0
ファイル: neo4j.py プロジェクト: rv816/graphlib
    def test_parse_create_rel(self):
        r = Node().relate(Node(), 'TO')
        s = 'MERGE (x0)-[x2:TO]->(x1)'
        # Third statement.. after creating the nodes
        self.assertEqual(neo4j.parse(serialize(r))[2], s)

        r['foo'] = 1
        r['bar'] = 'a'
        r.match_props = False
        s = "CREATE (x0)-[:TO {bar: 'a', foo: 1}]->(x1)"
        # Third statement.. after creating the nodes
        self.assertEqual(neo4j.parse(serialize(r))[2], s)
コード例 #8
0
ファイル: neo4j.py プロジェクト: bruth/graphlib
    def test_parse_create_node(self):
        n = Node({'foo': None})
        s = 'CREATE (x0)'
        self.assertEqual(neo4j.parse(serialize(n))[0], s)

        n['foo'] = 1
        n['bar'] = 'a'
        s = "CREATE (x0 {bar: 'a', foo: 1})"
        self.assertEqual(neo4j.parse(serialize(n))[0], s)

        n.labels = ['Special']
        s = "CREATE (x0:Special {bar: 'a', foo: 1})"
        self.assertEqual(neo4j.parse(serialize(n))[0], s)
コード例 #9
0
ファイル: neo4j.py プロジェクト: rv816/graphlib
    def test_parse_create_node(self):
        n = Node({'foo': None})
        s = 'CREATE (x0)'
        self.assertEqual(neo4j.parse(serialize(n))[0], s)

        n['foo'] = 1
        n['bar'] = 'a'
        s = "CREATE (x0 {bar: 'a', foo: 1})"
        self.assertEqual(neo4j.parse(serialize(n))[0], s)

        n.labels = ['Special']
        s = "CREATE (x0:Special {bar: 'a', foo: 1})"
        self.assertEqual(neo4j.parse(serialize(n))[0], s)
コード例 #10
0
ファイル: graph.py プロジェクト: rv816/graphlib
    def test_filter(self):
        items = Nodes([Node({'foo': 1}), Node({'foo': 2}), Node({'foo': 2})])

        self.assertCountEqual(items.filter('foo'), items)
        self.assertCountEqual(items.filter('foo', 1), items[:1])
        self.assertCountEqual(items.filter('foo', 2), items[1:])
        self.assertCountEqual(items.filter('foo', 3), [])
        self.assertCountEqual(items.filter('bar'), [])

        # Arbitrary filter function
        f = lambda n: n == items[2]
        self.assertCountEqual(items.filter(f), items[2:])

        # Only functions and strings
        self.assertRaises(TypeError, items.filter, 10)
コード例 #11
0
ファイル: neo4j.py プロジェクト: bruth/graphlib
    def setUp(self):
        n = Node({'foo': 'bar', 'baz': 10})

        # Add properties for coverage
        n.labels = ['Special']
        n.match_props = ['foo']
        n.update_props = ['baz']

        rels = []
        for i in range(5):
            rels.append(n.relate(Node({'index': i}), 'NEXT',
                        {'foo': i, 'bar': 2}))

        rels[0].match_props = ['foo']
        rels[0].update_props = ['bar']

        self.data = serialize(n)
コード例 #12
0
ファイル: graph.py プロジェクト: rv816/graphlib
    def test_parse_direction(self):
        s = Node()
        self.assertIsNone(s._parse_direction(incoming=True, outgoing=True))
        self.assertIsNone(s._parse_direction())

        self.assertEqual(s._parse_direction(direction=1), 1)
        self.assertEqual(s._parse_direction(outgoing=True), 1)
        self.assertEqual(s._parse_direction(outgoing=True), 1)
        self.assertEqual(s._parse_direction(incoming=True), -1)
コード例 #13
0
ファイル: graph.py プロジェクト: rv816/graphlib
    def test_relate_idempotent(self):
        s = Node()
        e = Node()

        r0 = s.relate(e, 'YES')
        r1 = s.relate(e, 'YES', {'one': 1})
        r2 = s.relate(e, 'YES')
        r3 = e.relate(s, 'YES', direction=-1)

        self.assertEqual(r1, r0)
        self.assertEqual(r2, r0)
        self.assertEqual(r3, r0)

        self.assertEqual(r0.props, {'one': 1})
コード例 #14
0
ファイル: graph.py プロジェクト: rv816/graphlib
    def test_relate_rev(self):
        s = Node()
        e = Node()

        r = e.relate(s, '-', direction=-1)
        _r = s.relate(e, '-', direction=1)
        self.assertEqual(r, _r)
コード例 #15
0
ファイル: neo4j.py プロジェクト: rv816/graphlib
    def test_parse_merge_node(self):
        n = Node({
            'foo': 1,
            'bar': 'a',
            'baz': None
        },
                 labels=['Special'],
                 match_props=['foo'])
        s = "MERGE (x0:Special {foo: 1}) " \
            "ON CREATE SET x0 = {bar: 'a', foo: 1} " \
            "ON MATCH SET x0.bar = 'a', x0.foo = 1"
        self.assertEqual(neo4j.parse(serialize(n))[0], s)

        # Replace properties on match
        d = serialize(n)
        d[0]['replace'] = True
        s = "MERGE (x0:Special {foo: 1}) " \
            "ON CREATE SET x0 = {bar: 'a', foo: 1} " \
            "ON MATCH SET x0 = {bar: 'a', foo: 1}"
        self.assertEqual(neo4j.parse(d)[0], s)
コード例 #16
0
ファイル: neo4j.py プロジェクト: rv816/graphlib
    def setUp(self):
        n = Node({'foo': 'bar', 'baz': 10})

        # Add properties for coverage
        n.labels = ['Special']
        n.match_props = ['foo']
        n.update_props = ['baz']

        rels = []
        for i in range(5):
            rels.append(
                n.relate(Node({'index': i}), 'NEXT', {
                    'foo': i,
                    'bar': 2
                }))

        rels[0].match_props = ['foo']
        rels[0].update_props = ['bar']

        self.data = serialize(n)
コード例 #17
0
ファイル: graph.py プロジェクト: rv816/graphlib
    def test_degree(self):
        n0 = Node()
        n1 = Node()
        n2 = Node()
        n3 = Node()

        self.assertEqual(n0.degree, 0)

        n0.relate(n1, 'X')
        self.assertEqual(n0.degree, 1)
        self.assertEqual(n1.degree, 1)

        n0.relate(n2, 'X')
        self.assertEqual(n0.degree, 2)
        self.assertEqual(n1.degree, 1)
        self.assertEqual(n2.degree, 1)

        n2.relate(n3, 'X')
        self.assertEqual(n0.degree, 2)
        self.assertEqual(n1.degree, 1)
        self.assertEqual(n2.degree, 2)
        self.assertEqual(n3.degree, 1)
コード例 #18
0
ファイル: graph.py プロジェクト: rv816/graphlib
 def test_init(self):
     self.assertEqual(Node([('one', 1)]).props, {'one': 1})
     self.assertEqual(Node({'one': 1}).props, {'one': 1})
     self.assertEqual(Node().props, {})
コード例 #19
0
ファイル: graph.py プロジェクト: rv816/graphlib
    def test_unrelate(self):
        s = Node()
        e = Node()
        o = Node()

        s.relate(e, 'A')
        s.relate(e, 'B')
        s.relate(e, 'C')
        s.relate(o, 'B')
        s.relate(o, 'D')

        self.assertEqual(s.unrelate(e, 'A'), 1)
        self.assertEqual(s.unrelate(e), 2)
        self.assertEqual(s.unrelate(type='B'), 1)
        self.assertEqual(s.unrelate(), 1)

        e.relate(s, 'A')
        s.relate(e, 'A')

        self.assertEqual(s.unrelate(e, direction=1), 1)
        self.assertEqual(s.unrelate(e, direction=-1), 1)
コード例 #20
0
ファイル: graph.py プロジェクト: rv816/graphlib
 def test_relate_multi(self):
     s = Node()
     o = Node()
     t = Node()
     s.relate([o, t], 'TRI')
コード例 #21
0
ファイル: graph.py プロジェクト: rv816/graphlib
    def test_rels(self):
        s = Node()
        e = Node()
        o = Node()

        r0 = s.relate(e, 'R')
        r1 = s.relate(e, 'A', direction=-1)
        r2 = s.relate(o, 'R')

        self.assertCountEqual(s.rels(e), [r0, r1])
        self.assertCountEqual(s.rels(e, direction=1), [r0])
        self.assertCountEqual(s.rels(e, direction=-1), [r1])

        self.assertCountEqual(s.rels(type='R'), [r0, r2])
        self.assertCountEqual(s.rels(type='R', direction=1), [r0, r2])
        self.assertCountEqual(s.rels(type='A', direction=-1), [r1])

        self.assertCountEqual(s.rels(e, type='R'), [r0])
        self.assertCountEqual(s.rels(e, type='R', direction=1), [r0])
        self.assertCountEqual(s.rels(e, type='A', direction=-1), [r1])

        self.assertCountEqual(s.rels(), [r0, r1, r2])
        self.assertCountEqual(s.rels(direction=1), [r0, r2])
        self.assertCountEqual(s.rels(direction=-1), [r1])
コード例 #22
0
ファイル: graph.py プロジェクト: rv816/graphlib
 def test_props(self):
     n = Node()
     n['one'] = 1
     self.assertEqual(n['one'], 1)
     del n['one']
     self.assertFalse('one' in n)
コード例 #23
0
ファイル: graph.py プロジェクト: rv816/graphlib
 def test_iter(self):
     self.assertEqual(list(Node({'key': 'value'})), ['key'])
コード例 #24
0
ファイル: graph.py プロジェクト: rv816/graphlib
    def test_relate(self):
        s = Node()
        e = Node()

        r = s.relate(e, 'YES', direction=1)
        _r = e.relate(s, 'YES', direction=-1)
        self.assertEqual(r, _r)

        # Any relationship
        self.assertTrue(s.related(e))
        self.assertTrue(e.related(s))

        # Specific relationship
        self.assertTrue(s.related(e, 'YES'))
        self.assertTrue(e.related(s, 'YES'))

        self.assertFalse(s.related(e, 'NO'))
        self.assertFalse(e.related(s, 'NO'))

        # Specific direction
        self.assertTrue(s.related(e, 'YES', outgoing=True))
        self.assertTrue(e.related(s, 'YES', incoming=True))

        self.assertFalse(s.related(e, 'YES', incoming=True))
        self.assertFalse(e.related(s, 'YES', outgoing=True))
コード例 #25
0
ファイル: serializer.py プロジェクト: rv816/graphlib
 def test_serialize(self):
     self.assertTrue(serialize(Node()))