def testFoundElement(self):
linac = Linac()
linac.addSection("TEST-abc")
linac.updateSvg()
zoomNode = self.svgFile.getZoom2Background()
sectionRectNode = self.svgFile.getElementById("section1bottomRect", zoomNode)
self.assertEqual(sectionRectNode.attrib["id"], "section1bottomRect")
self.assertEqual(sectionRectNode.attrib["width"], "630")
def testFoundElement(self):
linac = Linac()
linac.addSection("TEST-abc")
linac.updateSvg()
zoomNode = self.svgFile.getZoom2Background()
sectionRectNode = self.svgFile.getElementById("section1bottomRect",
zoomNode)
self.assertEqual(sectionRectNode.attrib["id"], "section1bottomRect")
self.assertEqual(sectionRectNode.attrib["width"], "630")
#!/usr/bin/python
import src.svg as svg
from src.Linac import Linac
from lxml import etree
from src.TangoDeviceManager import TangoDeviceManager
blankSVGpath = './src/blank.svg'
svgTree = etree.parse(blankSVGpath)
svgRoot = svgTree.getroot()
svgFile = svg.SVG()
svgFile.setSvg(svgRoot)
print "SVG file loaded "
L = Linac()
L.addSection("AB-01")
L.addSection("BB02")
L.addSection("CC03")
L.addLastLongSection()
L.updateSvg()
svgRoot = svgFile.getSvg()
toWrite = etree.tostring(svgRoot, pretty_print=True)
fd = open("new.svg", 'w')
fd.write(toWrite)
class TestLinac(unittest.TestCase):
def setUp(self):
blankSVGpath = './blank.svg'
svgTree = etree.parse(blankSVGpath)
svgRoot = svgTree.getroot()
self.svgFile = svg.SVG()
self.svgFile.setSvg(svgRoot)
self.linac = Linac()
def testSectionColour(self):
firstColour = self.linac.getNextSectionColour()
self.assertEqual(firstColour, "#55ffdd")
self.linac.addSection("AB01")
nextSectionColour = self.linac.getNextSectionColour()
self.assertEqual(nextSectionColour, "#b3b3b3")
self.linac.addSection("TEST")
nextSectionColour = self.linac.getNextSectionColour()
self.assertEqual(nextSectionColour, "grey")
def testNumberOfSections(self):
self.assertEqual(self.linac.numberOfSections(), 0)
self.linac.addSection("A")
self.assertEqual(self.linac.numberOfSections(), 1)
self.linac.addSection("B")
self.linac.addSection("C")
self.assertEqual(self.linac.numberOfSections(), 3)
def testGetSection(self):
self.assertEqual(self.linac.getSection("A"), None)
self.linac.addSection("A")
self.assertNotEqual(self.linac.getSection("A"), None)
sectionA = self.linac.sections[0]
self.assertEqual(self.linac.getSection("A"), sectionA)
self.linac.addSection("TEST-01")
sectionTest = self.linac.sections[1]
self.assertEqual(self.linac.getSection("TEST-01"), sectionTest)
def testAddSection(self):
self.assertEqual(self.linac.numberOfSections(), 0)
addedSection = self.linac.addSection("test002")
self.linac.updateSvg()
newSection = self.linac.getSection("test002")
self.assertEqual(newSection.longName, "test002")
self.assertEqual(addedSection, newSection)
zoomNode = self.svgFile.getZoom2Background()
bottomRect = self.svgFile.getElementById("section1bottomRect",
zoomNode)
self.assertEqual(bottomRect.attrib["x"], "1885")
self.assertEqual(bottomRect.attrib["y"], "3685")
self.assertEqual(self.linac.numberOfSections(), 1)
def testAddLastLongSection(self):
self.assertEqual(self.linac.numberOfSections(), 0)
self.linac.addSection("test001")
self.linac.addLastLongSection()
self.assertEqual(self.linac.numberOfSections(), 2)
firstSection = self.linac.getSection("test001")
lastSection = self.linac.getSection("")
self.assertNotEqual(lastSection, None)
self.assertEqual(firstSection.width, 630)
self.assertEqual(lastSection.width, self.linac.width - 630)
self.linac.updateSvg()
zoomNode = self.svgFile.getZoom2Background()
bottomRect = self.svgFile.getElementById("section2bottomRect",
zoomNode)
self.assertEqual(bottomRect.attrib["x"], "2515")
def testAddDevice(self):
device = Device("ABC/MAG/device123", None, None)
anotherDevice = Device("ABC/VAC/dev7", None, None)
self.linac.addSection("ABC")
section = self.linac.getSection("ABC")
self.assertEqual(section.numberOfDevices(), 0)
self.assertEqual(section.getDevice(0), None)
self.linac.addDevice(device)
self.assertEqual(section.getDevice(0), device)
self.assertEqual(section.getDevice(1), None)
self.linac.addDevice(anotherDevice)
self.assertEqual(section.getDevice(1), anotherDevice)
self.assertEqual(section.numberOfDevices(), 2)
def testAddDeviceToSubsection(self):
section = self.linac.addSection("I-S01", None, 200)
subsection = section.addSubsection("I-S01B", None, 50)
device = Device("I-S01B/ab/cd", None, [0, 20])
self.linac.addDevice(device)
self.assertEqual(device.section, subsection)
def testComputeCoordinate(self):
self.assertEqual(self.linac.numberOfSections(), 0)
self.assertEqual(self.linac.computeNewSectionCoordinate(), 1885)
self.linac.addSection("FOO")
self.assertEqual(self.linac.computeNewSectionCoordinate(), 2515)
def testAssignDevicesBeforeDrawing(self):
section = self.linac.addSection("I-K01", None, 200)
subsection = section.addSubsection("I-K01A", None, 50)
firstDevice = Device("I-K01/ab/cd", None, [12, 20])
secondDevice = Device("I-K01A/ef/gh", None, [5, 20])
thirdDevice = Device("I-K01A/ij/kl", None, [15, 20])
self.linac.addDevice(firstDevice)
self.linac.addDevice(secondDevice)
self.linac.addDevice(thirdDevice)
self.assertEqual(firstDevice.section, section)
self.linac.assignDevicesBeforeDrawing()
self.assertEqual(firstDevice.section, subsection)
def testGetAllDevicesSorted(self):
section = self.linac.addSection("A", None, 100)
subsection = section.addSubsection("A-1", None, 20)
firstDevice = Device("first", None, [301, 222])
secondDevice = Device("second", None, [154, 222])
thirdDevice = Device("third", None, [101, 222])
fourthDevice = Device("fourth", None, [55, 30])
subsection.addDevice(firstDevice)
subsection.addDevice(secondDevice)
section.addDevice(thirdDevice)
section.addDevice(fourthDevice)
devices = self.linac.getAllDevicesSorted()
self.assertEqual(
devices, [fourthDevice, thirdDevice, secondDevice, firstDevice])
self.assertEqual(devices[0].numberInSection, 0)
self.assertEqual(devices[2].numberInSection, 0)
self.assertEqual(devices[3].numberInSection, 1)
def tearDown(self):
pass
