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