def testAssigningSvgCoordinate(self): device = Device("abc/testDevice", None, None) self.assertEqual(device.svgCoordinateX, None) linac = Linac() linac.addSection("abc") linac.addDevice(device) device.assignSvgCoordinatesForLinacDevice() testSection = linac.getSection("abc") sectionStartCoordinate = testSection.startCoordinate distanceBetweenDevices = testSection.getDistanceBetweenDevices() self.assertEqual(device.svgCoordinateX, sectionStartCoordinate + distanceBetweenDevices)