def test_solutioninputtransformer_reset_solution(): """Does reset properly transform the text with solution=True?""" transformer = SolutionInputTransformer(True) for line in lines: transformer.push(line) output = transformer.reset() assert output == "# YOUR CODE HERE\nprint \"hello\"\n" assert transformer._lines == []
def test_solutioninputtransformer_reset_release(): """Does reset properly transform the text with solution=False?""" transformer = SolutionInputTransformer(False) for line in lines: transformer.push(line) output = transformer.reset() assert output == "# YOUR CODE HERE\nprint \"goodbye\"\n" assert transformer._lines == []
def test_solutioninputtransformer_push(): """Does push properly collect the lines?""" transformer = SolutionInputTransformer(True) for line in lines: transformer.push(line) assert transformer._lines == lines