def testSimple(self):
# bunch of datasets
datasets = [
Dataset(samples=N.arange(12).reshape((4,3)), labels=1),
Dataset(samples=N.zeros((4,4)), labels=1),
Dataset(samples=N.ones((4,2), dtype='float'), labels=1),
]
mds = MetaDataset(datasets)
# all together
self.failUnless(mds.samples.shape == (4, 9))
# should do upcasting
self.failUnless(mds.samples.dtype == 'float')
# simple samples attrs
self.failUnless((mds.labels == [1] * 4).all())
self.failUnless((mds.chunks == range(4)).all())
# do sample selection across all datasets
mds1 = mds.selectSamples([0,3])
self.failUnless(mds1.samples.shape == (2, 9))
self.failUnless(\
(mds1.samples[0] == [0, 1, 2, 0, 0, 0, 0, 1, 1]).all())
self.failUnless(\
(mds1.samples[1] == [9, 10, 11, 0, 0, 0, 0, 1, 1]).all())
# more tricky feature selection on all datasets
mds2 = mds.selectFeatures([1,4,8])
self.failUnless(\
(mds2.samples == [[ 1, 0, 1],
[ 4, 0, 1],
[ 7, 0, 1],
[10, 0, 1]] ).all())
