|
|
|
"""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!') |
|
|
