python competitive programming template for pytest

competitive.py

from sys import *


def main(cin, cout):
    a, b = map(int, cin.readline().split())
    cout.write(str(a + b))


def get_main_output(input_str):
    from io import StringIO
    cin = StringIO(input_str)
    cout = StringIO()
    main(cin, cout)
    return cout.getvalue()


def test_example():
    out = get_main_output('1 -2')
    assert out == '-1'


if __name__ == '__main__':
    main(stdin, stdout)