Exemplo n.º 1
0
    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)