From 28c2ae9102d4204b3f0a79419eec1e72dbbc529a Mon Sep 17 00:00:00 2001
From: David Luevano Alvarado <david@luevano.xyz>
Date: Tue, 21 Feb 2023 21:02:23 -0600
Subject: add configuration testing, small refactor

---
 tests/test_custom_logger.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

(limited to 'tests/test_custom_logger.py')

diff --git a/tests/test_custom_logger.py b/tests/test_custom_logger.py
index 1062e41..9102f52 100644
--- a/tests/test_custom_logger.py
+++ b/tests/test_custom_logger.py
@@ -1,6 +1,5 @@
 import pytest
-from logging import DEBUG, INFO, WARNING, ERROR, CRITICAL
-from typing import Callable
+from logging import Logger, DEBUG, INFO, WARNING, ERROR, CRITICAL
 
 
 @pytest.mark.parametrize('log_level, starts_with, message', [
@@ -13,7 +12,7 @@ from typing import Callable
 def test_log_levels(log_level: int,
                     starts_with: str,
                     message: str,
-                    logger: Callable,
+                    logger: Logger,
                     capture_stdout: dict[str, str | int]) -> None:
     logger.log(log_level, message)
     assert str(capture_stdout['stdout']).startswith(starts_with)
-- 
cgit v1.2.3-54-g00ecf