Commit c5c8a51a authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Incremented unit test for the Generator

parent 8989b45c
Pipeline #29569 failed with stage
in 69 minutes and 44 seconds
......@@ -95,10 +95,11 @@ def test_dataset_using_generator():
reader,\
multiple_samples=True)
iterator = dataset.make_one_shot_iterator().get_next()
while True:
for i in range(11):
try:
sample = session.run(iterator)
sample = session.run(iterator)
assert sample[0]["data"].shape == shape
assert numpy.allclose(sample[0]["data"], samples[i])
except tf.errors.OutOfRangeError:
break
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment