def test_iter(test_iter_args):
    reader = CSVInputReader()
    reader._CSVInputReader__head = ["from", "to", "amount"]
    reader._CSVInputReader__data = iter([test_iter_args[0]])
    iterator = iter(reader)
    n = iterator.next()
    assert type(n) == type(test_iter_args[1])
    assert vars(n) == vars(test_iter_args[1])
Example #2
0
def test_iter(test_iter_args):
    reader = CSVInputReader()
    reader._CSVInputReader__head = ['from', 'to', 'amount']
    reader._CSVInputReader__data = iter([test_iter_args[0]])
    iterator = iter(reader)
    n = iterator.next()
    assert type(n) == type(test_iter_args[1])
    assert vars(n) == vars(test_iter_args[1])