def test_DebugPrint_no_stack_frame(capsys, monkeypatch): # NOQA: N802
monkeypatch.setattr('jnrbase.debug.inspect.currentframe', lambda: None)
DebugPrint.enable()
try:
print('boom')
assert 'unknown:000] boom\n' in capsys.readouterr()[0]
finally:
DebugPrint.disable()
def test_DebugPrint(capsys): # NOQA: N802
DebugPrint.enable()
try:
print('boom')
out, _ = capsys.readouterr()
assert 'test_debug.py:' in out
assert '] boom\n' in out
finally:
DebugPrint.disable()
def test_DebugPrint(stdout):
DebugPrint.enable()
try:
print("boom")
out = stdout.getvalue()
expect(out).contains('test_debug.py:')
expect(out).contains('] boom\n')
finally:
DebugPrint.disable()
def test_DebugPrint_double_toggle(): # NOQA: N802
DebugPrint.enable()
sys.stdout.first = True
try:
DebugPrint.enable()
assert sys.stdout.first
finally:
DebugPrint.disable()
assert not hasattr(sys.stdout, 'first')
DebugPrint.disable()
def test_DebugPrint_double_enable():
DebugPrint.enable()
sys.stdout.first = True
try:
DebugPrint.enable()
expect(sys.stdout.first) is True
finally:
DebugPrint.disable()
