def test_teardown_logging(self): # create simple logger logger = logging.getLogger('a_logger') logger.addHandler(logging.StreamHandler()) stools.tearDownLogging('a_logger') self.assertEqual(len(logger.handlers), 0) del logger
def test_setup_logging(self): logger = stools.setUpLogging(verbose=True, debug=True, logname='a_logger') # more than two handlers: console + log.txt self.assertEqual(len(logger.handlers), 2) # debug level self.assertTrue(logger.level == 10) # a log file has been created self.assertTrue(os.path.isfile('log.txt')) # bring down the logger stools.tearDownLogging('a_logger') # remove log file os.remove('log.txt') del logger
def test_sort_args(self): # make sure error is raised if no names.txt directory = '.' email = 'an.email' logger = stools.setUpLogging(verbose=False, debug=False, logname='testlogger') with self.assertRaises(stools.PrimingError): stools.sortArgs(directory, email, logger) # create a names.txt and test arguments returned names = ['name1', 'name2', 'name3', 'name4', 'name5', 'name6'] with open('names.txt', 'w') as file: for name in names: file.write(name + '\n') res = stools.sortArgs(directory, email, logger) self.assertTrue(isinstance(res['terms'], list)) self.assertTrue(isinstance(res['genedict'], dict)) self.assertTrue(isinstance(res['paradict'], dict)) os.remove('names.txt') # bring down the logger stools.tearDownLogging('testlogger') # remove its log file os.remove('log.txt')