def test_add_fw(self): wf1 = VaspWorkflow(self.fw1) wf1.add_fw(self.fw2) wf1.add_fw(self.fw3) wf_dic = wf1.wf.as_dict() ldic = wf_dic['links'] # Rebuild WF1 Dictionary to reflect FWs '-1', '-2', and '-3' instead # of higher FW_ID numbers min_fw_id = int(min(ldic.keys())) modifier = -( min_fw_id + 1 ) link_dic = {str(min_fw_id + modifier): [ldic[str(min_fw_id)][0] + modifier], str(min_fw_id + modifier - 1): [ldic[str(min_fw_id-1)][0] + modifier], str(min_fw_id + modifier - 2): []} self.assertDictEqual(link_dic, self.links1, msg=None)