def test_can_get_sequence_pdbs_when_no_results(self, mock_xml_retriever, mock_json_retriever): mock_json_retriever.return_value = {"oneLetterSeq": "CCC"} mock_xml_retriever.return_value = None ligand = Ligand(self.ligand_json) pdbs = ligand.sequence_pdbs() self.assertEqual(pdbs, [])
def test_can_get_sequence_pdbs(self, mock_xml_retriever, mock_json_retriever): mock_json_retriever.return_value = {"oneLetterSeq": "CCC"} mock_xml_retriever.return_value = ["2XG3", "3A1I"] ligand = Ligand(self.ligand_json) pdbs = ligand.sequence_pdbs() self.assertEqual(pdbs, ["2XG3", "3A1I"])