File tree Expand file tree Collapse file tree 1 file changed +1
-13
lines changed Expand file tree Collapse file tree 1 file changed +1
-13
lines changed Original file line number Diff line number Diff line change 6
6
import unittest
7
7
from unittest .mock import MagicMock , patch
8
8
9
- from ads .aqua import configure_aqua_logger , get_logger_level , set_log_level
9
+ from ads .aqua import get_logger_level , set_log_level
10
10
11
11
12
12
class TestAquaLogging (unittest .TestCase ):
@@ -22,18 +22,6 @@ def test_get_logger_level_from_env(self):
22
22
"""Test log level is correctly read from environment variable."""
23
23
self .assertEqual (get_logger_level (), "DEBUG" )
24
24
25
- @patch ("logging.getLogger" )
26
- @patch ("logging.StreamHandler" )
27
- def test_configure_aqua_logger (self , mock_handler , mock_get_logger ):
28
- """Test that logger is correctly configured."""
29
- mock_logger = MagicMock ()
30
- mock_get_logger .return_value = mock_logger
31
-
32
- logger = configure_aqua_logger ()
33
-
34
- mock_get_logger .assert_called_once_with ("ads.aqua" )
35
- mock_logger .setLevel .assert_called_with (self .DEFAULT_AQUA_LOG_LEVEL )
36
-
37
25
@patch ("ads.aqua.logger" , create = True )
38
26
def test_set_log_level (self , mock_logger ):
39
27
"""Test that the log level of the logger is set correctly."""
You can’t perform that action at this time.
0 commit comments