pytest로 입출력을 테스트하는 방법

test_input_and_output.py

import csv
from io import StringIO

# 입력 스트림을 테스트, 출력   (스트링을 스트림으로 만들 수도 있음)
def test_load_csv():
    raw = StringIO('a,b,c\n1,2,3\n4,5,6') # string을 stream으로 변경
    actual = load_csv(raw)
    expected = [
        {'a': '1', 'b': '2', 'c': '3'},
        {'a': '4', 'b': '5', 'c': '6'},
    ]
    assert expected == actual

def load_csv(f):
    return list(csv.DictReader(f))