def test_no_stdout():
temp = sys.stdout
sys.stdout = None
logger = tasklogger.TaskLogger("no stdout")
logger.info("no stdout")
logger.logger.handlers[0].stream.flush()
sys.stdout = temp
def test_log():
logger = tasklogger.TaskLogger("test_log")
logger.debug('debug')
logger.info('info')
logger.warning('warning')
logger.error('error')
logger.critical('critical')
def test_log():
logger = tasklogger.TaskLogger("test_log")
logger.debug("debug")
logger.info("info")
logger.warning("warning")
logger.error("error")
logger.critical("critical")
def test_tasks():
logger = tasklogger.TaskLogger("test_tasks")
logger.start_task("test")
assert time.time() - logger.tasks['test'] < 0.01
time.sleep(logger.min_runtime)
logger.complete_task("test")
assert 'test' not in logger.tasks
logger.complete_task("another test")
def test_custom_timer():
logger = tasklogger.TaskLogger("test_custom_timer")
logger.set_timer(lambda: 1000 * time.time())
logger.start_task("test")
time.sleep(logger.min_runtime)
runtime = logger.complete_task("test")
assert runtime is not None
assert runtime >= 1000 * logger.min_runtime
def test_cpu_timer():
logger = tasklogger.TaskLogger("test_cpu_timer")
logger.set_timer("cpu")
logger.start_task("test")
time.sleep(logger.min_runtime * 10)
runtime = logger.complete_task("test")
assert runtime is not None
assert runtime < logger.min_runtime * 10
def test_ipynb():
def monkey_patch():
return True
temp = tasklogger.utils.in_ipynb
tasklogger.utils.in_ipynb = monkey_patch
logger = tasklogger.TaskLogger("ipynb")
logger.info("ipynb")
tasklogger.utils.in_ipynb = temp
def test_oserror():
def monkey_patch(*args, **kwargs):
raise OSError("[Errno 9] Bad file descriptor")
temp = os.write
os.write = monkey_patch
logger = tasklogger.TaskLogger("oserror")
logger.info("oserror")
os.write = temp
def test_tasks():
logger = tasklogger.TaskLogger("test_tasks")
logger.start_task("test")
assert time.time() - logger.tasks["test"] < 0.01
time.sleep(logger.min_runtime)
runtime = logger.complete_task("test")
assert runtime is not None
assert runtime >= logger.min_runtime
assert "test" not in logger.tasks
logger.complete_task("another test")
def test_level():
logger = tasklogger.TaskLogger("test_level")
logger.set_level(2)
assert logger.level == logging.DEBUG
assert logger.logger.level == logging.DEBUG
logger.set_level(1)
assert logger.level == logging.INFO
assert logger.logger.level == logging.INFO
logger.set_level(0)
assert logger.level == logging.WARNING
assert logger.logger.level == logging.WARNING
def test_exists():
logger = tasklogger.TaskLogger("test_exists")
with np.testing.assert_raises(RuntimeError):
tasklogger.TaskLogger("test_exists")
logger_1 = tasklogger.TaskLogger("test_exists", if_exists="increment")
assert logger_1.name == "test_exists_1"
logger_2 = tasklogger.TaskLogger("test_exists", if_exists="increment")
assert logger_2.name == "test_exists_2"
logger_3 = tasklogger.TaskLogger("test_exists", if_exists="ignore")
assert logger_3.logger == logger.logger
with np.testing.assert_raises(ValueError):
tasklogger.TaskLogger("test_exists", if_exists="bad_value")
def test_context():
logger = tasklogger.TaskLogger("test_context_api")
with tasklogger.log_task("test", logger="test_context_api"):
assert "test" in logger.tasks
assert "test" not in logger.tasks
def test_duplicate():
logger = tasklogger.TaskLogger("test_duplicate")
np.testing.assert_raises(RuntimeError, tasklogger.TaskLogger,
"test_duplicate")
logger2 = tasklogger.TaskLogger("test_no_duplicate")
assert logger.logger is not logger2.logger
def test_context():
logger = tasklogger.TaskLogger("test_context")
with logger.task("test"):
assert "test" in logger.tasks
assert "test" not in logger.tasks
def test_context():
logger = tasklogger.TaskLogger("test_context_api")
with tasklogger.log_task('test', logger='test_context_api'):
assert 'test' in logger.tasks
assert 'test' not in logger.tasks
def test_stderr():
temp = sys.stdout
sys.stdout = None
logger = tasklogger.TaskLogger("stderr", stream="stderr")
logger.info("stderr")
sys.stdout = temp
def test_context():
logger = tasklogger.TaskLogger("test_context")
with logger.task('test'):
assert 'test' in logger.tasks
assert 'test' not in logger.tasks
def test_bad_timer():
logger = tasklogger.TaskLogger("test_bad_timer")
np.testing.assert_raises(ValueError, logger.set_timer, "bad")
def test_indent():
logger = tasklogger.TaskLogger("test_indent")
logger.set_indent(0)
assert logger.indent == 0
