diff --git a/ippisite/ippidb/tests.py b/ippisite/ippidb/tests.py index 2c2d02a6d01dd840c602d6b62c201f2761ca1865..3119c4816838d37d445ecfde0c62d1217c4fd3c7 100644 --- a/ippisite/ippidb/tests.py +++ b/ippisite/ippidb/tests.py @@ -11,8 +11,20 @@ from django.urls import reverse import requests -from ippidb import ws, models -from ippidb.ws import get_uniprot_info +from ippidb import models +from ippidb.ws import ( + get_uniprot_info, + get_doi_info, + get_pfam_info, + get_google_patent_info, + PatentNotFound, + get_pubmed_info, + get_pdb_uniprot_mapping, + get_pdb_pfam_mapping, + EntryNotFoundError, + convert_iupac_to_smiles_and_inchi, + convert_smiles_to_iupac, +) from .models import ( Compound, CompoundTanimoto, @@ -29,7 +41,7 @@ from .models import DrugBankCompound, Protein class MolSmiTestCase(TestCase): """ - Test MOL to SMILES and SMILES to MOL format conversion functions + Test MOL to SMILES and SMILES to MOL format conversion iPPI-DB web services """ def setUp(self): @@ -538,7 +550,7 @@ class TestGetDoiInfo(TestCase): def test_get_doi_info(self): try: - resp = ws.get_doi_info("10.1073/pnas.0805139105") + resp = get_doi_info("10.1073/pnas.0805139105") except requests.exceptions.HTTPError as he: # skip this test if the DOI server throws an error # (that happens) @@ -561,7 +573,7 @@ class TestGetDoiInfo(TestCase): class TestGetPfamInfo(TestCase): def test_get_pfam_info(self): target = {"id": "bZIP_1", "description": "bZIP transcription factor"} - resp = ws.get_pfam_info("PF00170") + resp = get_pfam_info("PF00170") self.assertEqual(resp, target) @@ -576,13 +588,13 @@ class TestGetGooglePatentInfo(TestCase): " McGrew, Robert T. Bell, " "Steven Joseph Rich, Cisco Technology Inc", } - resp = ws.get_google_patent_info("US8856504") + resp = get_google_patent_info("US8856504") self.assertEqual(resp, target) def test_entry_not_found(self): self.assertRaises( - ws.PatentNotFound, - ws.get_google_patent_info, + PatentNotFound, + get_google_patent_info, "US8856504US8856504US885US8856504US8856504", ) @@ -608,7 +620,7 @@ class TestGetPubMEDIdInfo(TestCase): "authors_list": "Brancotte B, Biton A, Bernard-Pierrot I, " "Radvanyi F, Reyal F, Cohen-Boulakia S", } - resp = ws.get_pubmed_info("21349868") + resp = get_pubmed_info("21349868") self.assertEqual(resp, target) @@ -730,13 +742,13 @@ class TestGetPDBUniProtMapping(TestCase): def test_find_info(self): target = sorted(["Q03164", "O00255"]) - resp = ws.get_pdb_uniprot_mapping("3u85") + resp = get_pdb_uniprot_mapping("3u85") resp = sorted(resp) self.assertEqual(resp, target) self.assertEqual(len(resp), len(set(resp))) def test_entry_not_found(self): - self.assertRaises(ws.EntryNotFoundError, ws.get_pdb_uniprot_mapping, "Xu85") + self.assertRaises(EntryNotFoundError, get_pdb_uniprot_mapping, "Xu85") class TestGetPDBPfamMapping(TestCase): @@ -748,11 +760,11 @@ class TestGetPDBPfamMapping(TestCase): target = { "PF05053": {"identifier": "Menin", "description": "Menin", "name": "Menin"} } - resp = ws.get_pdb_pfam_mapping("3u85") + resp = get_pdb_pfam_mapping("3u85") self.assertDictEqual(resp, target) def test_entry_not_found(self): - self.assertRaises(ws.EntryNotFoundError, ws.get_pdb_uniprot_mapping, "Xu85") + self.assertRaises(EntryNotFoundError, get_pdb_pfam_mapping, "Xu85") class TestConvertIUPACToSMILESAndMore(TestCase): @@ -792,7 +804,7 @@ class TestConvertIUPACToSMILESAndMore(TestCase): ), ] for iupac, dict_expected in pairs: - dict_returned = ws.convert_iupac_to_smiles_and_inchi(iupac) + dict_returned = convert_iupac_to_smiles_and_inchi(iupac) self.assertEqual(dict_expected["smiles"], dict_returned["smiles"]) self.assertEqual(dict_expected["inchi"], dict_returned["inchi"]) self.assertEqual(dict_expected["stdinchi"], dict_returned["stdinchi"]) @@ -800,12 +812,12 @@ class TestConvertIUPACToSMILESAndMore(TestCase): def test_invalid_entry(self): self.assertRaises( - ws.EntryNotFoundError, - ws.convert_iupac_to_smiles_and_inchi, + EntryNotFoundError, + convert_iupac_to_smiles_and_inchi, "3-{1-oxo-6-[4-(piperid", ) self.assertRaises( - ws.EntryNotFoundError, ws.convert_iupac_to_smiles_and_inchi, None + EntryNotFoundError, convert_iupac_to_smiles_and_inchi, None ) @@ -818,7 +830,7 @@ class TestConvertSMILESToIUPAC(TestCase): smiles_to_iupacs = {"CCC": "propane"} for smiles, expected_iupac in smiles_to_iupacs.items(): self.assertEqual( - ws.convert_smiles_to_iupac(smiles).lower(), expected_iupac.lower() + convert_smiles_to_iupac(smiles).lower(), expected_iupac.lower() )