def test_can_get_gtop_pdbs_by_species(self, mock_json_retriever): mock_json_retriever.return_value = self.pdb_json target = Target(self.target_json) pdbs = target.gtop_pdbs(species="rat") self.assertEqual(pdbs, ["4IAR"])
def test_gtop_pdbs_when_no_json(self, mock_json_retriever): mock_json_retriever.return_value = None target = Target(self.target_json) self.assertEqual(target.gtop_pdbs(), [])