File size: 1,108 Bytes
8c212a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
# python3.7
"""Unit test for logger."""
import time
from .logger import build_logger
def test_logger():
"""Test function."""
for logger_type in ['normal', 'rich', 'dumb']:
if logger_type == 'normal':
class_name = 'Logger'
elif logger_type == 'rich':
class_name = 'RichLogger'
elif logger_type == 'dumb':
class_name = 'DumbLogger'
print(f'===== Test `utils.logger.{class_name}` =====')
logger = build_logger(logger_type,
logger_name=logger_type,
logfile_name=f'test_{logger_type}_logger.log')
logger.print('print log')
logger.debug('debug log')
logger.info('info log')
logger.warning('warning log')
logger.init_pbar()
task1 = logger.add_pbar_task('Task 1', 500)
task2 = logger.add_pbar_task('Task 2', 1000)
for _ in range(1000):
logger.update_pbar(task1, 1)
logger.update_pbar(task2, 1)
time.sleep(0.005)
logger.close_pbar()
print('Success!')
|