def test_unpack():
data = range(10)
stream = Batch(DataStream(IterableDataset(data)),
iteration_scheme=ConstantScheme(2))
wrapper = Unpack(stream)
epoch = wrapper.get_epoch_iterator()
for i, v in enumerate(epoch):
assert numpy.shape(v)[0] == 1
assert v[0] == i
def test_unpack():
data = range(10)
stream = Batch(
DataStream(IterableDataset(data)), iteration_scheme=ConstantScheme(2))
wrapper = Unpack(stream)
epoch = wrapper.get_epoch_iterator()
for i, v in enumerate(epoch):
assert numpy.shape(v)[0] == 1
assert v[0] == i
def test_unpack_picklable(self):
wrapper = Unpack(self.stream_np)
epoch = wrapper.get_epoch_iterator()
cPickle.dumps(epoch)
def test_unpack(self):
wrapper = Unpack(self.stream)
epoch = wrapper.get_epoch_iterator()
for i, v in enumerate(epoch):
assert numpy.shape(v)[0] == 1
assert v[0] == i
def test_unpack_picklable(self):
wrapper = Unpack(self.stream_np)
epoch = wrapper.get_epoch_iterator()
cPickle.dumps(epoch)
def test_unpack(self):
wrapper = Unpack(self.stream)
epoch = wrapper.get_epoch_iterator()
for i, v in enumerate(epoch):
assert numpy.shape(v)[0] == 1
assert v[0] == i
