Fix unittest for activating pysnooper

This commit is contained in:
pikez 2019-06-18 23:40:00 +08:00 committed by Ram Rachum
parent b4c8c16ed9
commit d7d3a80c16
1 changed files with 3 additions and 3 deletions

View File

@ -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()