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)
示例#2
0
    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)
示例#4
0
    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)
示例#6
0
    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)
示例#9
0
 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"))
示例#10
0
    def testComputeSubsectionCoordinate(self):
        section = LinacSection("test", None, 100)
        self.assertEqual(section.computeSubsectionStartCoordinate(), 100)

        section.addSubsection("subtest", None, 45)
        self.assertEqual(section.computeSubsectionStartCoordinate(), 145)