def test_can_get_name_pdbs_when_no_results(self, mock_xml_retriever): mock_xml_retriever.return_value = None ligand = Ligand(self.ligand_json) pdbs = ligand.name_pdbs() self.assertEqual(pdbs, [])
def test_can_get_name_pdbs(self, mock_xml_retriever): mock_xml_retriever.return_value = ["2XG3", "3A1I"] ligand = Ligand(self.ligand_json) pdbs = ligand.name_pdbs() self.assertEqual(pdbs, ["2XG3", "3A1I"])