def testGetDevice(self): dev1 = Device("foo", None, None) dev2 = Device("bar", None, None) section = Section("test", "grey") self.assertEqual(section.numberOfDevices(), 0) section.addDevice(dev1) section.addDevice(dev2) self.assertEqual(section.numberOfDevices(), 2) self.assertEqual(section.getDevice(0), dev1) self.assertNotEqual(section.getDevice(0), dev2) self.assertEqual(section.getDevice(1), dev2) self.assertEqual(section.getDevice(3), None)
def testSortDevices(self): firstDevice = Device("First", None, [184, 99]) secondDevice = Device("Second", None, [964, 80]) section = Section("test", None) section.addDevice(secondDevice) section.addDevice(firstDevice) self.assertEqual(section.getDevice(0), secondDevice) self.assertEqual(section.getDevice(0).numberInSection, 0) self.assertEqual(section.getDevice(1).numberInSection, 1) section.sortDevices() self.assertEqual(section.getDevice(0), firstDevice) self.assertEqual(section.getDevice(0).numberInSection, 0) self.assertEqual(section.getDevice(1).numberInSection, 1)