def test_splitstatevector(self):
testdir = self.cppqeddir
for name in os.listdir(testdir):
readpath = os.path.join(testdir, name)
evs, qs = io.load_cppqed(readpath)
tempdirpath = tempfile.mkdtemp(prefix="pycppqed_test_")
writepath = os.path.join(tempdirpath, "cppqed")
io.split_cppqed(readpath, writepath)
evs2, qs2 = io.load_cppqed(writepath)
svnames = os.listdir(tempdirpath)
svnames.sort()
svs2 = [io.load_statevector(os.path.join(tempdirpath, svname))\
for svname in svnames[1:]]
shutil.rmtree(tempdirpath)
self.assert_((evs2==evs).all())
svs2 = statevector.StateVectorTrajectory(svs2)
self.assert_((svs2==qs.statevector).all())
def test_loadcppqed(self):
testdir = self.cppqeddir
for name in os.listdir(testdir):
path = os.path.join(testdir, name)
evs, qs = io.load_cppqed(path)