def testShortName(self): firstDevice = Device("I-K01/VAC/I-K01CAB05-VAC-IPCU1", None, None) secondDevice = Device("R1-SGA/MAG/R1-SGACAB14-MAG-PS04", None, None) self.assertEqual(firstDevice.getShortName(), "IPCU1") self.assertEqual(secondDevice.getShortName(), "PS04")