def testAddDevice(self): device = Device("abcde", None, None) section = Section("test", "grey") self.assertEqual(section.numberOfDevices(), 0) section.addDevice(device) self.assertEqual(section.numberOfDevices(), 1) self.assertEqual(section.devices[0], device) self.assertEqual(device.numberInSection, 0) otherDevice = Device("other", None, None) section.addDevice(otherDevice) self.assertEqual(otherDevice.numberInSection, 1)
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)