Show More
@@ -23,7 +23,11 b' def transform_checker(tests, transformer):' | |||||
23 | transformer = transformer() |
|
23 | transformer = transformer() | |
24 | try: |
|
24 | try: | |
25 | for inp, tr in tests: |
|
25 | for inp, tr in tests: | |
26 | nt.assert_equal(transformer.push(inp), tr) |
|
26 | if inp is None: | |
|
27 | out = transformer.reset() | |||
|
28 | else: | |||
|
29 | out = transformer.push(inp) | |||
|
30 | nt.assert_equal(out, tr) | |||
27 | finally: |
|
31 | finally: | |
28 | transformer.reset() |
|
32 | transformer.reset() | |
29 |
|
33 |
General Comments 0
You need to be logged in to leave comments.
Login now