def test_can_get_gtop_pdbs(self, mock_json_retriever): self.interaction_json["ligandId"] = 149 mock_json_retriever.side_effect = [[self.interaction_json], self.pdb_json] ligand = Ligand(self.ligand_json) pdbs = ligand.gtop_pdbs() self.assertEqual(pdbs, ["4IAR"])
def test_gtop_pdbs_when_no_json(self, mock_json_retriever): mock_json_retriever.return_value = None ligand = Ligand(self.ligand_json) self.assertEqual(ligand.gtop_pdbs(), [])