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
