def testNewSaveImage(self): a = Node(x=40,y=240,color='#00FF00',label='Point A') b = Node(x=600,y=240,color='#00FF00',label='Point B') m = Map([Link(a,b,1000,width=3,debug=True)]) m.draw_arrows() m.draw_labels() m.save('map.png')
class TestNode(unittest.TestCase): def setUp(self): self.a_node = Node(x=40,y=240,color='#00FF00',label='Point A') self.b_node = Node(x=340,y=240,color='#00FF00',label='Point B') self.map = Map([Link(self.a_node,self.b_node,1000,width=3,debug=True)]) def testDrawArrows(self): self.map.draw_arrows() def testDrawLabels(self): self.map.draw_labels() def testSaveImage(self): self.map.save('map.png') def testNewSaveImage(self): a = Node(x=40,y=240,color='#00FF00',label='Point A') b = Node(x=600,y=240,color='#00FF00',label='Point B') m = Map([Link(a,b,1000,width=3,debug=True)]) m.draw_arrows() m.draw_labels() m.save('map.png')