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