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 66import unittest
77from unittest .mock import MagicMock , patch
88
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
1010
1111
1212class TestAquaLogging (unittest .TestCase ):
@@ -22,18 +22,6 @@ def test_get_logger_level_from_env(self):
2222 """Test log level is correctly read from environment variable."""
2323 self .assertEqual (get_logger_level (), "DEBUG" )
2424
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-
3725 @patch ("ads.aqua.logger" , create = True )
3826 def test_set_log_level (self , mock_logger ):
3927 """Test that the log level of the logger is set correctly."""
You can’t perform that action at this time.
0 commit comments