Skip to content

Commit 6697725

Browse files
committed
Revert "[MINOR][TESTS] Use SparkContext.setLogLevel in QuietTest"
This reverts commit 9bab16a.
1 parent 1a502d3 commit 6697725

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

python/pyspark/testing/utils.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -194,16 +194,14 @@ def wrapper(*args: Any, **kwargs: Any) -> Any:
194194

195195
class 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

209207
class PySparkTestCase(unittest.TestCase):

0 commit comments

Comments
 (0)