fix pfam test for pfam changes

jamesmkrieger · Feb 7, 2024 · 4bab30f · 4bab30f
1 parent 9ef920b
commit 4bab30f
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/prody/tests/database/test_pfam.py b/prody/tests/database/test_pfam.py
@@ -36,9 +36,8 @@ def testUniprotAccMulti(self):
         self.assertIsInstance(a, dict,
             'searchPfam failed to return a dict instance')
 
-        self.assertEqual(sorted(list(a.keys())), 
-                           ['PF00060', 'PF00497', 'PF01094', 'PF10613'],
-                           'searchPfam failed to return the right domain family IDs')
+        self.assertEqual(sorted(list(a.keys()))[:2], ['PF00060', 'PF00497'],
+                         'searchPfam failed to return the right domain family IDs')
 
     def testPdbIdChMulti(self):
         """Test the outcome of a simple search scenario using a PDB ID
@@ -49,7 +48,7 @@ def testPdbIdChMulti(self):
         self.assertIsInstance(a, dict,
             'searchPfam failed to return a dict instance')
 
-        self.assertEqual(sorted(list(a.keys())), ['PF00060', 'PF00497', 'PF01094', 'PF10613'],
+        self.assertEqual(sorted(list(a.keys()))[:2], ['PF00060', 'PF00497'],
                          'searchPfam failed to return the right domain family IDs for AMPAR')
 
     def testPdbIdChSingle(self):