def test_iter(test_iter_args): reader = TXTInputReader() reader._TXTInputReader__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])
def test_open(): reader = TXTInputReader() assert reader._TXTInputReader__data == None reader.open(123) assert reader._TXTInputReader__data == 123