def test_to_json():
bs = BasisSet.from_str(VTZGAUSSIAN, fmt='gaussian', name='VTZ')
bsdumped = bs.to_json(indent=4)
bsloaded = BasisSet.from_json(bsdumped)
assert bs.name == bsloaded.name, 'inconsistent name'
assert bs.element == bsloaded.element, 'inconsistent element'
for shell, funs in bs.functions.items():
assert shell in bsloaded.functions.keys(), 'missing shell {}'.format(shell)
assert np.allclose(funs['e'], bsloaded.functions[shell]['e'])
for f1, f2 in zip(funs['cf'], bsloaded.functions[shell]['cf']):
assert np.allclose(f1['idx'], f2['idx']), 'inconsistent idx'
assert np.allclose(f1['cc'], f2['cc']), 'inconsistent cc'
def bsconvert():
'''
CLI script to convert between different basis set formats
'''
parser = ArgumentParser(
description='Convert basis set between formats of different programs')
parser.add_argument("filename",
help="file name with a basis set, default='pickle'")
parser.add_argument(
"-from",
"--inputformat",
choices=["gamessus", "gaussian", "json", "molpro", "pickle"],
help="Basis set input format",
default="pickle")
parser.add_argument("-to",
"--outputformat",
choices=[
"cfour", "dalton", "gamessus", "gaussian", "json",
"molpro", "nwchem", "pickle"
],
help="Basis set output format",
default="molpro")
parser.add_argument('-o', '--output', help='name of the output file')
args = parser.parse_args()
name = os.path.splitext(args.filename)[0]
if args.inputformat == "pickle":
bsets = BasisSet.from_pickle(args.filename)
elif args.inputformat == "json":
bsets = BasisSet.from_json(args.filename)
else:
bsets = BasisSet.from_file(fname=args.filename,
fmt=args.inputformat,
name=name)
if args.outputformat == "pickle":
if isinstance(bsets, dict):
for elem, bset in bsets.items():
bset.to_pickle(name + '_' + elem + '.pkl')
elif isinstance(bsets, BasisSet):
bsets.to_pickle(name + '_' + bsets.element + '.pkl')
else:
writer_name = "to_" + args.outputformat
writer_objs = []
if isinstance(bsets, dict):
for elem, bset in bsets.items():
writer_objs.append(getattr(bset, writer_name))
elif isinstance(bsets, BasisSet):
writer_objs.append(getattr(bsets, writer_name))
else:
raise ValueError('Something went wrong')
if args.output:
fobj = open(args.output, 'w')
else:
fobj = sys.stdout
for writer in writer_objs:
print(writer(), file=fobj)