import contextlib
import io

from example3 import print_table
from capture_stdout import capture_stdout

def test_sum():
    with capture_stdout() as output:
        print_table()
    assert 'sum: 115' in output.getvalue()