def setUp(self):
rng = numpy.random.RandomState(123)
self.stream = DataStream(IndexableDataset(OrderedDict([
('features', rng.rand(4, 2, 2)),
('targets', numpy.array([0, 1, 0, 1]))
]),
axis_labels={
'features':
('batch', 'width',
'height'),
'targets': ('batch', )
}),
iteration_scheme=SequentialScheme(4, 2))
self.duplicate = Duplicate(self.stream, 'features')
def test_init(self):
duplicate = Duplicate(self.stream)
assert_equal(duplicate.which_sources, self.stream.sources)
assert_equal(self.duplicate.which_sources, ['features'])
