def should_work_end_to_end():
input = ['1 2', '3 4', '5 6 7']
expected = ['1 2', '2 1',
'3 4', '4 3',
'5 6 7', '6 5 7', '7 5 6']
with mock_console_io(tabify('\n'.join(input))) as (out, _):
main()
assert out.getvalue() == tabify('\n'.join(expected) + '\n')
def should_handle_extended_network():
input = tabify('''1 2 # 3
2 1 3 # 4
3 2 4 # 1 5
4 3 5 # 2 6''')
with mock_console_io(input) as (out, _):
main()
assert out.getvalue() == tabify('''1 2 # 3
2 # 1 3
2 1 3 # 4
1 # 2 3 4
3 # 2 1 4
3 2 4 # 1 5
2 # 3 4 1 5
4 # 3 2 1 5
4 3 5 # 2 6
3 # 4 5 2 6
5 # 4 3 2 6
''')
def should_handle_extended_network():
input = tabify('''1 2 # 3
2 1 3 # 4
3 2 4 # 1 5
4 3 5 # 2 6''')
with mock_console_io(input) as (out, _):
main()
assert out.getvalue() == tabify('''1 2 # 3
2 # 1 3
2 1 3 # 4
1 # 2 3 4
3 # 2 1 4
3 2 4 # 1 5
2 # 3 4 1 5
4 # 3 2 1 5
4 3 5 # 2 6
3 # 4 5 2 6
5 # 4 3 2 6
''')
def should_work_end_to_end():
input = ['1 2', '3 4', '5 6 7']
expected = ['1 2', '2 1', '3 4', '4 3', '5 6 7', '6 5 7', '7 5 6']
with mock_console_io(tabify('\n'.join(input))) as (out, _):
main()
assert out.getvalue() == tabify('\n'.join(expected) + '\n')