add test protein

KarrLab · Feb 14, 2019 · 17cb233 · 17cb233
1 parent afab339
commit 17cb233
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/alphabet/test_protein.py b/tests/alphabet/test_protein.py
@@ -11,6 +11,7 @@
 from wc_utils.util.chem import EmpiricalFormula
 import os.path
 import shutil
+import requests
 import tempfile
 import unittest
 
@@ -40,6 +41,14 @@ def test_ProteinForm_properties(self):
         self.assertEqual(form.get_formula(), EmpiricalFormula('C9H17N3O4'))
         self.assertEqual(form.get_charge(), 3 * monomers.A.get_charge())
 
+    def test_ProteinForm_get_monomer_details(self):
+        path = os.path.join(self.dirname, 'alphabet.yml')
+        session = requests.Session()
+        self.assertEqual(protein.ProteinAlphabetBuilder().get_monomer_details('AA0037', session)[0], 'S')
+        self.assertEqual(protein.ProteinAlphabetBuilder().get_monomer_details('AA0037', session)[3], {'AA0016'})
+        # add more for the identifiers
+
+
     def test_ProteinAlphabetBuilder(self):
         path = os.path.join(self.dirname, 'alphabet.yml')