def test_octagon(self):
"""Test octagon nodes."""
data = interwiki_graph.Subject(self.pages['en'])
data.found_in[self.pages['en']] = [self.pages['fr'], self.pages['pl']]
data.found_in[self.pages['en2']] = [self.pages['fr']]
data.found_in[self.pages['fr']] = [self.pages['en'], self.pages['pl']]
data.found_in[self.pages['pl']] = [self.pages['en'], self.pages['fr']]
drawer = interwiki_graph.GraphDrawer(data)
self.assertEqual({self.pages['en'].site}, drawer._octagon_site_set())
drawer.createGraph()
nodes = drawer.graph.obj_dict['nodes']
self.assertEqual(
nodes['"pl:origin"'][0]['attributes']['shape'],
'rectangle')
self.assertEqual(
nodes['"fr:origin"'][0]['attributes']['shape'],
'rectangle')
self.assertEqual(
nodes['"en:origin"'][0]['attributes']['shape'],
'octagon')
def test_simple_graph(self):
"""Test that GraphDrawer.createGraph does not raise exception."""
data = interwiki_graph.Subject(self.pages['en'])
data.found_in[self.pages['en']] = [self.pages['fr'], self.pages['pl']]
data.found_in[self.pages['fr']] = [self.pages['en'], self.pages['pl']]
data.found_in[self.pages['pl']] = [self.pages['en'], self.pages['fr']]
drawer = interwiki_graph.GraphDrawer(data)
drawer.createGraph()
def test_octagon(self):
"""Test octagon nodes."""
self.data.found_in[self.pages['en2']] = [self.pages['fr']]
drawer = interwiki_graph.GraphDrawer(self.data)
self.assertEqual({self.pages['en'].site}, drawer._octagon_site_set())
drawer.createGraph()
nodes = drawer.graph.obj_dict['nodes']
for node, shape in [('"pl:origin"', 'rectangle'),
('"fr:origin"', 'rectangle'),
('"en:origin"', 'octagon')]:
with self.subTest(node=node):
self.assertEqual(nodes[node][0]['attributes']['shape'], shape)
def test_simple_graph(self):
"""Test that GraphDrawer.createGraph does not raise exception."""
drawer = interwiki_graph.GraphDrawer(self.data)
drawer.createGraph()
