def test_read_matdyn():
# matdyn.freq
kpoints, freqs = pwscf.read_matdyn_freq('files/matdyn.freq')
aae(kpoints, kpoints_ref)
aae(freqs, freqs_ref)
# matdyn modes: read_matdyn_modes()
qpoints, freqs, vecs = pwscf.read_matdyn_modes('files/matdyn.modes',
natoms=2)
aae(qpoints, qpoints_ref)
aae(freqs, freqs_modes_ref)
aae(vecs.real, vecs_real_ref)
aaae(vecs.imag, vecs_imag_ref)
def test_read_all_dyn():
# matdyn modes: read_all_dyn()
qpoints, freqs, vecs = pwscf.read_all_dyn('files/dyn/', nqpoints=2,
natoms=2, base='ph.dyn')
aae(qpoints, qpoints_ref)
aae(freqs, freqs_modes_ref)
aae(vecs.real, vecs_real_ref)
aaae(vecs.imag, vecs_imag_ref)
def test_rpdf():
have_vmd = os.system('which vmd > /dev/null 2>&1') == 0
for name in ['rand_3d', 'aln_ibrav0_sc', 'aln_ibrav2_sc']:
print(("name: %s" % name))
dd = 'files/rpdf'
if name == 'rand_3d':
# 2 Trajectory = 2 selections
cell = np.loadtxt(pj(dd, name + '.cell.txt'))
coords_frac = [
load_old(pj(dd, name + '.coords0.txt')),
load_old(pj(dd, name + '.coords1.txt'))
]
trajs = [
crys.Trajectory(coords_frac=cf, cell=cell)
for cf in coords_frac
]
for tr in trajs:
assert tr.coords_frac.shape == (20, 10, 3)
assert tr.nstep == 20
assert tr.natoms == 10
else:
# one Structure
struct = parse.CifFile(pj(dd, name + '.cif')).get_struct()
trajs = [struct]
cell = struct.cell
ret = crys.rpdf(trajs, rmax=5.0, dr=0.05, pbc=True)
# rpdf() -- compere w/ ref
results = {
'rad': ret[:, 0],
'hist': ret[:, 1],
'num_int': ret[:, 2],
'rmax_auto': np.array(crys.rmax_smith(cell)),
}
for key, val in results.items():
print((" key: %s" % key))
ref_fn = pj(dd, "result.%s.%s.txt" % (key, name))
print((" reference file: %s" % ref_fn))
ref = np.loadtxt(ref_fn)
if doplot:
plt.figure()
plt.plot(ref, '.-', label='ref')
plt.plot(val, '.-', label='val')
plt.legend()
plt.title(key)
else:
# decimal=3 b/c ref data created w/ older implementation,
# slight numerical noise
np.testing.assert_array_almost_equal(ref, val, decimal=3)
print((" key: %s ... ok" % key))
# API
if name.startswith('aln_'):
sy = np.array(trajs[0].symbols)
ret1 = crys.rpdf(trajs, dr=0.1, amask=[sy == 'Al', sy == 'N'])
ret2 = crys.rpdf(trajs, dr=0.1, amask=['Al', 'N'])
aae(ret1, ret2)
# API:
# [traj, traj]
# [traj]
# traj
traj = crys.Trajectory(coords_frac=rand(100, 20, 3),
cell=np.identity(3) * 20,
symbols=['O'] * 5 + ['H'] * 15)
ret1 = crys.rpdf([traj, traj], rmax=5.0, dr=0.05, pbc=True)
ret2 = crys.rpdf([traj], rmax=5.0, dr=0.05, pbc=True)
aae(ret1, ret2)
ret3 = crys.rpdf(traj, rmax=5.0, dr=0.05, pbc=True)
aae(ret1, ret3)
# dmask
ret = crys.rpdf(traj, rmax=5.0, dr=0.05, dmask='>=2.0')
msk = ret[:, 0] >= 2.0
imsk = np.invert(msk)
assert (ret[msk, 1] > 0.0).any()
assert (ret[imsk, 1] == 0.0).all()
ret = crys.rpdf(traj, rmax=5.0, dr=0.05, dmask='{d}>=2.0')
assert (ret[msk, 1] > 0.0).any()
assert (ret[imsk, 1] == 0.0).all()
ret = crys.rpdf(traj,
rmax=5.0,
dr=0.05,
dmask='({d} >= 1.0) & ({d} <= 3.0)')
msk = (ret[:, 0] >= 1.0) & (ret[:, 0] <= 3.0)
imsk = np.invert(msk)
assert (ret[msk, 1] > 0.0).any()
assert (ret[imsk, 1] == 0.0).all()
if have_vmd:
# slicefirst and API
print("vmd_measure_gofr: slicefirst ...")
traj = crys.Trajectory(coords_frac=rand(100, 20, 3),
cell=np.identity(3) * 20,
symbols=['O'] * 5 + ['H'] * 15)
for first, last, step in [(0, -1, 1), (20, 80, 10)]:
ret = []
for sf in [True, False]:
print("first=%i, last=%i, step=%i, slicefirst=%s" %
(first, last, step, sf))
tmp = crys.vmd_measure_gofr(
traj,
dr=0.1,
rmax='auto',
sel=['all', 'all'],
slicefirst=sf,
first=first,
last=last,
step=step,
usepbc=1,
tmpdir=testdir,
verbose=False,
)
ret.append(tmp)
assert np.allclose(ret[0][:, 0], ret[1][:, 0])
assert np.allclose(ret[0][:, 1], ret[1][:, 1])
assert np.allclose(ret[0][:, 2], ret[1][:, 2])
# API call_vmd_measure_gofr()
trajfn = pj(testdir, 'vmd_xsf_call_vmd_measure_gofr')
data = io.write_axsf(trajfn, traj)
ret = crys.call_vmd_measure_gofr(trajfn,
dr=0.1,
rmax=10,
sel=['all', 'all'],
fntype='xsf',
first=0,
last=-1,
step=1,
usepbc=1,
datafn=None,
scriptfn=None,
logfn=None,
tmpdir=testdir,
verbose=False)
# compare results, up to L/2 = rmax_auto = 10 = rmax_smith(cell)
# all-all, hist will differ
rmax = 10
vmd = crys.vmd_measure_gofr(traj, dr=0.1, sel=['all', 'all'], rmax=10)
pwt = crys.rpdf(traj, dr=0.1, amask=None, rmax=10)
assert np.allclose(vmd[:-1, 0], pwt[:, 0]) # rad
##assert np.allclose(vmd[:-1,1], pwt[:,1]) # hist
assert np.allclose(vmd[:-1, 2], pwt[:, 2]) # num_int
# 2 selections, all ok
sy = np.array(traj.symbols)
vmd = crys.vmd_measure_gofr(traj,
dr=0.1,
sel=['name O', 'name H'],
rmax=10)
pwt = crys.rpdf(traj, dr=0.1, amask=[sy == 'O', sy == 'H'], rmax=10)
assert np.allclose(vmd[:-1, 0], pwt[:, 0]) # rad
assert np.allclose(vmd[:-1, 1], pwt[:, 1]) # hist
assert np.allclose(vmd[:-1, 2], pwt[:, 2]) # num_int
if doplot:
plt.show()
def test_rpdf():
have_vmd = os.system("which vmd > /dev/null 2>&1") == 0
for name in ["rand_3d", "aln_ibrav0_sc", "aln_ibrav2_sc"]:
print ("name: %s" % name)
dd = "files/rpdf"
if name == "rand_3d":
# 2 Trajectory = 2 selections
cell = np.loadtxt(pj(dd, name + ".cell.txt"))
coords_frac = [load_old(pj(dd, name + ".coords0.txt")), load_old(pj(dd, name + ".coords1.txt"))]
trajs = [crys.Trajectory(coords_frac=cf, cell=cell) for cf in coords_frac]
for tr in trajs:
assert tr.coords_frac.shape == (20, 10, 3)
assert tr.nstep == 20
assert tr.natoms == 10
else:
# one Structure
struct = parse.CifFile(pj(dd, name + ".cif")).get_struct()
trajs = [struct]
cell = struct.cell
ret = crys.rpdf(trajs, rmax=5.0, dr=0.05, pbc=True)
# rpdf() -- compere w/ ref
results = {
"rad": ret[:, 0],
"hist": ret[:, 1],
"num_int": ret[:, 2],
"rmax_auto": np.array(crys.rmax_smith(cell)),
}
for key, val in results.iteritems():
print (" key: %s" % key)
ref_fn = pj(dd, "result.%s.%s.txt" % (key, name))
print (" reference file: %s" % ref_fn)
ref = np.loadtxt(ref_fn)
if doplot:
plt.figure()
plt.plot(ref, ".-", label="ref")
plt.plot(val, ".-", label="val")
plt.legend()
plt.title(key)
else:
# decimal=3 b/c ref data created w/ older implementation,
# slight numerical noise
np.testing.assert_array_almost_equal(ref, val, decimal=3)
print (" key: %s ... ok" % key)
# API
if name.startswith("aln_"):
sy = np.array(trajs[0].symbols)
ret1 = crys.rpdf(trajs, dr=0.1, amask=[sy == "Al", sy == "N"])
ret2 = crys.rpdf(trajs, dr=0.1, amask=["Al", "N"])
aae(ret1, ret2)
# API:
# [traj, traj]
# [traj]
# traj
traj = crys.Trajectory(coords_frac=rand(100, 20, 3), cell=np.identity(3) * 20, symbols=["O"] * 5 + ["H"] * 15)
ret1 = crys.rpdf([traj, traj], rmax=5.0, dr=0.05, pbc=True)
ret2 = crys.rpdf([traj], rmax=5.0, dr=0.05, pbc=True)
aae(ret1, ret2)
ret3 = crys.rpdf(traj, rmax=5.0, dr=0.05, pbc=True)
aae(ret1, ret3)
# dmask
ret = crys.rpdf(traj, rmax=5.0, dr=0.05, dmask=">=2.0")
msk = ret[:, 0] >= 2.0
imsk = np.invert(msk)
assert (ret[msk, 1] > 0.0).any()
assert (ret[imsk, 1] == 0.0).all()
ret = crys.rpdf(traj, rmax=5.0, dr=0.05, dmask="{d}>=2.0")
assert (ret[msk, 1] > 0.0).any()
assert (ret[imsk, 1] == 0.0).all()
ret = crys.rpdf(traj, rmax=5.0, dr=0.05, dmask="({d} >= 1.0) & ({d} <= 3.0)")
msk = (ret[:, 0] >= 1.0) & (ret[:, 0] <= 3.0)
imsk = np.invert(msk)
assert (ret[msk, 1] > 0.0).any()
assert (ret[imsk, 1] == 0.0).all()
if have_vmd:
# slicefirst and API
print "vmd_measure_gofr: slicefirst ..."
traj = crys.Trajectory(coords_frac=rand(100, 20, 3), cell=np.identity(3) * 20, symbols=["O"] * 5 + ["H"] * 15)
for first, last, step in [(0, -1, 1), (20, 80, 10)]:
ret = []
for sf in [True, False]:
print "first=%i, last=%i, step=%i, slicefirst=%s" % (first, last, step, sf)
tmp = crys.vmd_measure_gofr(
traj,
dr=0.1,
rmax="auto",
sel=["all", "all"],
slicefirst=sf,
first=first,
last=last,
step=step,
usepbc=1,
tmpdir=testdir,
verbose=False,
)
ret.append(tmp)
assert np.allclose(ret[0][:, 0], ret[1][:, 0])
assert np.allclose(ret[0][:, 1], ret[1][:, 1])
assert np.allclose(ret[0][:, 2], ret[1][:, 2])
# API call_vmd_measure_gofr()
trajfn = pj(testdir, "vmd_xsf_call_vmd_measure_gofr")
data = io.write_axsf(trajfn, traj)
ret = crys.call_vmd_measure_gofr(
trajfn,
dr=0.1,
rmax=10,
sel=["all", "all"],
fntype="xsf",
first=0,
last=-1,
step=1,
usepbc=1,
datafn=None,
scriptfn=None,
logfn=None,
tmpdir=testdir,
verbose=False,
)
# compare results, up to L/2 = rmax_auto = 10 = rmax_smith(cell)
# all-all, hist will differ
rmax = 10
vmd = crys.vmd_measure_gofr(traj, dr=0.1, sel=["all", "all"], rmax=10)
pwt = crys.rpdf(traj, dr=0.1, amask=None, rmax=10)
assert np.allclose(vmd[:-1, 0], pwt[:, 0]) # rad
##assert np.allclose(vmd[:-1,1], pwt[:,1]) # hist
assert np.allclose(vmd[:-1, 2], pwt[:, 2]) # num_int
# 2 selections, all ok
sy = np.array(traj.symbols)
vmd = crys.vmd_measure_gofr(traj, dr=0.1, sel=["name O", "name H"], rmax=10)
pwt = crys.rpdf(traj, dr=0.1, amask=[sy == "O", sy == "H"], rmax=10)
assert np.allclose(vmd[:-1, 0], pwt[:, 0]) # rad
assert np.allclose(vmd[:-1, 1], pwt[:, 1]) # hist
assert np.allclose(vmd[:-1, 2], pwt[:, 2]) # num_int
if doplot:
plt.show()
