def testHasSubsections(self): section = LinacSection("I-K00", None, None) self.assertEqual(section.hasSubsections(), False) sectionWithSubsections = LinacSection("I-S01", None, 100) sectionWithSubsections.addSubsection("I-S01A", None, 0) self.assertEqual(sectionWithSubsections.hasSubsections(), True)
def testGetAllDevices(self): section = LinacSection("I-K00", None, 0) firstDevice = Device("test1", None, None) secondDevice = Device("test2", None, None) thirdDevice = Device("test3", None, None) fourthDevice = Device("test4", None, None) section.addSubsection("I-K01A", None, 100) section.addSubsection("I-K01B", None, 100) firstSubsection = section.getSubsection("I-K01A") secondSubsection = section.getSubsection("I-K01B") firstSubsection.addDevice(firstDevice) section.addDevice(secondDevice) secondSubsection.addDevice(thirdDevice) firstSubsection.addDevice(fourthDevice) allDevices = section.getAllDevices() self.assertTrue(firstDevice in allDevices) self.assertTrue(secondDevice in allDevices) self.assertTrue(thirdDevice in allDevices) self.assertTrue(fourthDevice in allDevices) self.assertEqual(len(allDevices), 4)
def testSortDevicesRecursively(self): section = LinacSection("A", None, 100) subsection = section.addSubsection("A01", None, 50) 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 = section.getAllDevices() self.assertEqual(devices[0], thirdDevice) self.assertEqual(devices[2], firstDevice) section.sortDevicesRecursively() devices = section.getAllDevices() self.assertEqual(devices[0], fourthDevice) self.assertEqual(devices[2], secondDevice)
def testSortDevicesRecursively(self): section = LinacSection("A", None, 100) subsection = section.addSubsection("A01", None, 50) 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 = section.getAllDevices() self.assertEqual(devices[0], thirdDevice) self.assertEqual(devices[2], firstDevice) section.sortDevicesRecursively() devices = section.getAllDevices() self.assertEqual(devices[0], fourthDevice) self.assertEqual(devices[2], secondDevice)
def testGetSubsection(self): section = LinacSection("I-K00", None, 100) subsection = section.addSubsection("I-K00A", None, 100) self.assertEqual(section.getSubsection("I-K00B"), None) self.assertEqual(section.getSubsection("I-K00A"), section.subsections[0]) self.assertEqual(subsection, section.getSubsection("I-K00A"))
def testComputeSubsectionCoordinate(self): section = LinacSection("test", None, 100) self.assertEqual(section.computeSubsectionStartCoordinate(), 100) section.addSubsection("subtest", None, 45) self.assertEqual(section.computeSubsectionStartCoordinate(), 145)