def test_array_reset(): array_ds = Stream('a', [1, 2, 3]) assert array_ds.next() == {'a': 1} assert array_ds.next() == {'a': 2} assert array_ds.next() == {'a': 3} array_ds.reset() assert array_ds.next() == {'a': 1} assert array_ds.next() == {'a': 2} assert array_ds.next() == {'a': 3}
def test_array_next(): array_ds = Stream('a', [1, 2, 3]) next_value = array_ds.next() assert next_value == {'a': 1}