File tree Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Original file line number Diff line number Diff line change @@ -194,16 +194,14 @@ def wrapper(*args: Any, **kwargs: Any) -> Any:
194194
195195class QuietTest :
196196 def __init__ (self , sc ):
197- self .sc = sc
197+ self .log4j = sc . _jvm . org . apache . log4j
198198
199199 def __enter__ (self ):
200- self .old_level = (
201- self .sc ._jvm .org .apache .log4j .LogManager .getRootLogger ().getLevel ().toString ()
202- )
203- self .sc .setLogLevel ("FATAL" )
200+ self .old_level = self .log4j .LogManager .getRootLogger ().getLevel ()
201+ self .log4j .LogManager .getRootLogger ().setLevel (self .log4j .Level .FATAL )
204202
205203 def __exit__ (self , exc_type , exc_val , exc_tb ):
206- self .sc . setLogLevel (self .old_level )
204+ self .log4j . LogManager . getRootLogger (). setLevel (self .old_level )
207205
208206
209207class PySparkTestCase (unittest .TestCase ):
You can’t perform that action at this time.
0 commit comments