def test_tag(self): logger = Logger(tee=False, indentation=4) logger.log("Message 1", Logger.DEBUG, tag="TEST") logger.log("Message 2", Logger.DEBUG) logger.log("Message 3", Logger.DEBUG, tag="TEST") logger.log("Message 4", Logger.DEBUG) strings = logger.to_list_of_strings() self.assertEqual(strings[0].find("TEST") > -1, True) self.assertEqual(strings[1].find("TEST") > -1, False) self.assertEqual(strings[2].find("TEST") > -1, True) self.assertEqual(strings[3].find("TEST") > -1, False)