build: handling flaky PMC test

biosimulators · Feb 27, 2022 · 9a64a1a · 9a64a1a
1 parent 511f252
commit 9a64a1a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/tests/ref/test_ref_utils.py b/tests/ref/test_ref_utils.py
@@ -1,6 +1,7 @@
 from biosimulators_utils.ref import utils
 from unittest import mock
 import Bio.Entrez
+import flaky
 import ftplib
 import os
 import random
@@ -45,6 +46,7 @@ def test_get_entrez_record(self):
         with self.assertRaisesRegex(TypeError, 'must be a string'):
             utils.get_entrez_record('pmc', None)
 
+    @flaky.flaky(max_runs=5, min_passes=1)
     def test_get_pubmed_central_id(self):
         self.assertEqual(utils.get_pubmed_central_id('23184105'), 'PMC5813803')
         time.sleep(1)

diff --git a/tests/requirements.txt b/tests/requirements.txt
@@ -1,2 +1,3 @@
 capturer
+flaky
 jsonschema