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'])