def test_add_non_directed_edge_with_multiple_edges_and_tuples(self):
g = Grafellow()
g.add_non_directed_edge('zach', 'person', 'laura', 'person',
edge_type='friends')
g.add_non_directed_edge('zach', 'person', 'jaco', 'robot',
edge_type='friends')
self.assertTrue(('jaco','robot') in g.node_edges('zach', 'person'))
self.assertTrue(('laura','person') in g.node_edges('zach', 'person'))
def test_retrieving_edge_type(self):
g = Grafellow(testing=True)
g.add_non_directed_edge('zach', 'person', 'laura', 'person',
edge_type='friends')
g.add_non_directed_edge('zach', 'person', 'jaco', 'robot',
edge_type='friends')
results = g.node_edges('zach', 'person', edge_types=True)
self.assertTrue(('friends', 'jaco', 'robot') in results)
self.assertTrue(('friends', 'laura', 'person') in results)
def test_add_non_directed_edge(self):
g = Grafellow()
g.add_non_directed_edge('zach', 'person', 'laura', 'person', edge_type='friends')
self.assertTrue(g.get_node('zach','person') == {})
self.assertTrue(g.get_node('laura','person') == {})
self.assertTrue('laura' in g.node_edges('zach', 'person', other_node_type='person'))
