def test_summary_report_custom(df, capfd, lst, expected): summary_report(df, lst) output = capfd.readouterr()[0].split("\n") assert output[0].strip() == expected
def test_summary_report(df, capfd, line, expected): summary_report(df) output = capfd.readouterr()[0].split("\n") assert output[line].strip() == expected