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)