From d7d3a80c16ec98345645e575046b9affad93a76c Mon Sep 17 00:00:00 2001 From: pikez Date: Tue, 18 Jun 2019 23:40:00 +0800 Subject: [PATCH] Fix unittest for activating pysnooper --- tests/test_pysnooper.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_pysnooper.py b/tests/test_pysnooper.py index e451caf..52b8a2e 100644 --- a/tests/test_pysnooper.py +++ b/tests/test_pysnooper.py @@ -1183,11 +1183,11 @@ def test_activate_deactivate_snoop(): y = 8 return x + y - os.environ['PYSNOOPER_DISABLED'] = '1' - with pysnooper.snoop(): + pysnooper.tracer.DISABLED = '1' + with pysnooper.snoop(string_io): result = my_function('baba') output = string_io.getvalue() assert output == "" - os.environ['PYSNOOPER_DISABLED'] = '' + pysnooper.tracer.DISABLED = '' test_string_io()