Mercurial > pylearn
diff _test_dataset.py @ 8:d1c394486037
Replaced asarray() method by __array__ method which gets called automatically when
trying to cast into an array or numpy array or upon numpy.asarray(dataset).
| author | bengioy@bengiomac.local |
|---|---|
| date | Mon, 24 Mar 2008 15:56:53 -0400 |
| parents | 6f8f338686db |
| children | be128b9127c8 |
line wrap: on
line diff
--- a/_test_dataset.py Mon Mar 24 13:20:15 2008 -0400 +++ b/_test_dataset.py Mon Mar 24 15:56:53 2008 -0400 @@ -16,23 +16,25 @@ a=ArrayDataSet(data=numpy.random.rand(8,3),fields={"x":slice(2),"y":slice(1,3)},minibatch_size=1) s=0 for example in a: - print len(example), example.x s+=_sum_all(example.x) - print s + #print s self.failUnless(abs(s-7.25967597)<1e-6) def test1(self): a=ArrayDataSet(data=numpy.random.rand(10,4),fields={"x":slice(2),"y":slice(1,4)},minibatch_size=1) a.minibatch_size=2 - print a.asarray() + s=0 for mb in a: - print mb,mb.asarray() - print "a.y=",a.y + s+=_sum_all(numpy.array(mb)) + s+=a[3:6].x[1,1] for mb in ArrayDataSet(data=a.y,minibatch_size=2): - print mb for e in mb: - print e - self.failUnless(True) + s+=sum(e) + #print numpy.array(a) + #print a.y[4:9:2] + s+= _sum_all(a.y[4:9:2]) + #print s + self.failUnless(abs(s-39.0334797)<1e-6) if __name__ == '__main__': unittest.main()