Improving/Expanding Functional Tests

stas · January 10, 2019, 9:25pm

FYI, added CaptureStdout context manager for a much more compact stdout capture and clearing:

fastai/fastai/blob/master/tests/utils/text.py#L18


# or contextlib.redirect_stdout) contains any temporary printed strings, followed by
# \r's. This helper function ensures that the buffer will contain the same output
# with and without -s in pytest, by turning:
# foo bar\r tar mar\r final message
# into:
# final message
# it can handle a single string or a multiline buffer
def apply_print_resets(buf):
return re.sub(r'^.*\r', '', buf, 0, re.M)


class CaptureStdout():
""" Context manager to capture stdout, clean it up and make it available via obj.out or str(obj).


Example:


with CaptureStdout() as cs:
    print("Secret message")
print(f"captured: {cs.out}")
# or via its stringified repr:
print(f"captured: {cs}")