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_saveloadstatevector(self):
SV = statevector.StateVector
a = SV((1,2,3), time=1)
b = SV((0,2+1j,5-2j,6), time=1.2)
pars = (
a,
b,
a^b,
a^b^a,
a^a^a^a,
)
for sv in pars:
f, path = tempfile.mkstemp(prefix="pycppqed_test_")
io.save_statevector(path, sv)
sv2 = io.load_statevector(path)
os.remove(path)
self.assert_((sv==sv2).all())
