def test_converge_wc(fresh_aiida_env, potentials, mock_vasp):
"""Test submitting only, not correctness, with mocked vasp code."""
from aiida.orm import Code
from aiida.plugins import WorkflowFactory
from aiida.engine import run
workchain = WorkflowFactory('vasp.converge')
mock_vasp.store()
create_authinfo(computer=mock_vasp.computer, store=True)
structure = PoscarParser(
file_path=data_path('test_converge_wc', 'inp', 'POSCAR')).structure
parameters = IncarParser(
file_path=data_path('test_converge_wc', 'inp', 'INCAR')).incar
parameters['system'] = 'test-case:test_converge_wc'
parameters = {
k: v
for k, v in parameters.items()
if k not in ['isif', 'ibrion', 'encut', 'nsw']
}
restart_clean_workdir = get_data_node('bool', False)
restart_clean_workdir.store()
inputs = AttributeDict()
inputs.code = Code.get_from_string('mock-vasp@localhost')
inputs.structure = structure
inputs.parameters = get_data_node('dict', dict={'incar': parameters})
inputs.potential_family = get_data_node('str', POTCAR_FAMILY_NAME)
inputs.potential_mapping = get_data_node('dict', dict=POTCAR_MAP)
inputs.options = get_data_node('dict',
dict={
'withmpi': False,
'queue_name': 'None',
'resources': {
'num_machines': 1,
'num_mpiprocs_per_machine': 1
},
'max_wallclock_seconds': 3600
})
inputs.max_iterations = get_data_node('int', 1)
inputs.clean_workdir = get_data_node('bool', False)
relax = AttributeDict()
converge = AttributeDict()
converge.relax = get_data_node('bool', False)
converge.compress = get_data_node('bool', False)
converge.displace = get_data_node('bool', False)
converge.pwcutoff_samples = get_data_node('int', 3)
converge.k_samples = get_data_node('int', 3)
relax.perform = get_data_node('bool', True)
inputs.relax = relax
inputs.converge = converge
inputs.verbose = get_data_node('bool', True)
results, node = run.get_node(workchain, **inputs)
assert node.exit_status == 0
converge = results['converge']
assert 'data' in converge
conv_data = converge['data']
try:
conv_data.get_array('pw_regular')
except KeyError:
pytest.fail('Did not find pw_regular in converge.data')
try:
conv_data.get_array('kpoints_regular')
except KeyError:
pytest.fail('Did not find kpoints_regular in converge.data')
assert 'pwcutoff_recommended' in converge
try:
_encut = converge['pwcutoff_recommended'].value
except AttributeError:
pytest.fail('pwcutoff_recommended does not have the expected format')
assert 'kpoints_recommended' in converge
try:
_kpoints = converge['kpoints_recommended'].get_kpoints_mesh()
except AttributeError:
pytest.fail('kpoints_recommended does not have the expected format')
def test_converge_wc_pw(fresh_aiida_env, vasp_params, potentials, mock_vasp):
"""Test convergence workflow using mock code."""
from aiida.orm import Code
from aiida.plugins import WorkflowFactory
from aiida.engine import run
workchain = WorkflowFactory('vasp.converge')
mock_vasp.store()
create_authinfo(computer=mock_vasp.computer).store()
structure = PoscarParser(file_path=data_path('test_converge_wc/pw/200',
'inp', 'POSCAR')).structure
parameters = IncarParser(
file_path=data_path('test_converge_wc/pw/200', 'inp', 'INCAR')).incar
parameters['system'] = 'test-case:test_converge_wc'
parameters = {
k: v
for k, v in parameters.items()
if k not in ['isif', 'ibrion', 'encut', 'nsw']
}
kpoints = KpointsParser(file_path=data_path('test_converge_wc/pw/200',
'inp', 'KPOINTS')).kpoints
restart_clean_workdir = get_data_node('bool', False)
restart_clean_workdir.store()
inputs = AttributeDict()
inputs.code = Code.get_from_string('mock-vasp@localhost')
inputs.structure = structure
inputs.kpoints = kpoints
inputs.parameters = get_data_node('dict', dict={'incar': parameters})
inputs.potential_family = get_data_node('str', POTCAR_FAMILY_NAME)
inputs.potential_mapping = get_data_node('dict', dict=POTCAR_MAP)
inputs.options = get_data_node('dict',
dict={
'withmpi': False,
'queue_name': 'None',
'resources': {
'num_machines': 1,
'num_mpiprocs_per_machine': 1
},
'max_wallclock_seconds': 3600
})
inputs.max_iterations = get_data_node('int', 1)
inputs.clean_workdir = get_data_node('bool', False)
relax = AttributeDict()
converge = AttributeDict()
relax.perform = get_data_node('bool', False)
converge.relax = get_data_node('bool', False)
converge.testing = get_data_node('bool', True)
converge.compress = get_data_node('bool', False)
converge.displace = get_data_node('bool', False)
converge.pwcutoff_samples = get_data_node('int', 3)
converge.k_samples = get_data_node('int', 3)
inputs.relax = relax
inputs.converge = converge
inputs.verbose = get_data_node('bool', True)
results, node = run.get_node(workchain, **inputs)
assert node.exit_status == 0
assert 'converge' in results
converge = results['converge']
assert 'data' in converge
conv_data = converge['data']
try:
conv_data = conv_data.get_array('pw_regular')
except KeyError:
pytest.fail('Did not find pw_regular in converge.data')
conv_data_test = np.array([[200.0, -10.77974998, 0.0, 0.0, 0.5984],
[250.0, -10.80762044, 0.0, 0.0, 0.5912],
[300.0, -10.82261992, 0.0, 0.0, 0.5876]])
np.testing.assert_allclose(conv_data, conv_data_test)
assert 'pwcutoff_recommended' in converge
try:
_encut = converge['pwcutoff_recommended'].value
np.testing.assert_equal(_encut, 300)
except AttributeError:
pytest.fail('pwcutoff_recommended does not have the expected format')
