Function test2

recipes/Python/361168_Vpython_example/recipe-361168.py:262–303

()

    a.quit()

def test2():
    from StringIO import StringIO
    oldStdin = sys.stdin
    testInput = '\n'.join([
        '10',
        'move 9 onto 1',
        'move 8 over 1',
        'move 7 over 1',
        'move 6 over 1',
        'pile 8 over 6',
        'pile 8 over 5',
        'move 2 over 1',
        'move 4 over 9',
        'pile 9 onto 7',
        'pile 9 onto 1',
        'quit',
        ''])
    sys.stdin = StringIO(testInput)
    oldStdout = sys.stdout
    sys.stdout = s = StringIO()
    parse()
    sys.stdin = oldStdin
    sys.stdout = oldStdout
    reqd = '\n'.join([
        ' 0: 0',
        ' 1: 1 9 2 4',
        ' 2: ',
        ' 3: 3',
        ' 4: ',
        ' 5: 5 8 7 6',
        ' 6: ',
        ' 7: ',
        ' 8: ',
        ' 9: ',
        ''])
    s = s.getvalue()
    if s != reqd:
        print repr(s)
        print repr(reqd)
    else:
        print 'OK'
    sys.exit()

if __name__ == '__main__':
    test2()