def test_cblitz(self):
if io.cio is None:
raise Exception("Can't test c extension!")
for a in self.arrays:
blitzstr1 = io._numpy2blitz(a)
na1 = io._blitz2numpy(blitzstr1)
blitzstr2 = io._numpy2blitz(statevector.StateVector(a))
na2 = io._blitz2numpy(blitzstr2)
self.assert_((a==na1).all())
self.assert_((a==na2).all())
def test_blitz(self):
cio = io.cio
io.cio = None
for a in self.arrays:
blitzstr1 = io._numpy2blitz(a)
na1 = io._blitz2numpy(blitzstr1)
blitzstr2 = io._numpy2blitz(statevector.StateVector(a))
na2 = io._blitz2numpy(blitzstr2)
self.assert_((numpy.abs(a-na1)<eps).all())
self.assert_((numpy.abs(a-na2)<eps).all())
io.cio = cio
def test_loadblitz(self):
basedir = os.path.dirname(os.path.abspath(__file__))
testdir = os.path.join(basedir, "test/blitzarray")
names = os.listdir(testdir)
names.sort()
for i, name in enumerate(names):
path = os.path.join(testdir, name)
f = open(path)
blitzstr = f.read()
f.close()
array = io._blitz2numpy(blitzstr)
self.assertEqual(array.ndim, i+1)
