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)