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()