def testGetAngleBetweenDevices(self): ringASection = RingAbstractSection("RAS", None, 20, 90) self.assertEqual(ringASection.getAngleBetweenDevices(), 90) ringASection.addDevice(Device("a", None, None)) self.assertEqual(ringASection.getAngleBetweenDevices(), 45) ringASection.addDevice(Device("b", None, None)) self.assertEqual(ringASection.getAngleBetweenDevices(), 30)
def testSortDevices(self): D1 = Device("D1", None, [0,1]) D2 = Device("D2", None, [1, 0]) D3 = Device("D3", None, [0, -1]) D4 = Device("D4", None, [-1,0 ]) section = RingAbstractSection("R1-01", "green", 0, 360) section.centerX = 0.0 section.centerY = 0.0 section.addDevice(D2) section.addDevice(D4) section.addDevice(D1) section.addDevice(D3) self.assertEqual(section.devices[0], D2) self.assertEqual(section.devices[3], D3) section.sortDevices() self.assertEqual(section.devices[0], D1) self.assertEqual(section.devices[3], D4)
def testIsInUpperHalf(self): upperSection = RingAbstractSection("InUpper", None, -30, 180) lowerSection = RingAbstractSection("InLower", None, 100, 120) self.assertEqual(upperSection.isInUpperHalf(), True) self.assertEqual(lowerSection.isInUpperHalf(), False)