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(), [])
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(), [])