test: fix for pypy

Author: lepture

tests/test_security_edge_cases.py

@@ -10,16 +10,15 @@ def test_blank_list_continuations_are_near_linear(self):
 
         def render(size):
             text = "- first\n" + "\n" * size + "  last\n"
-            started = time.process_time()
+            started = time.perf_counter()
             md(text)
-            return time.process_time() - started
+            return time.perf_counter() - started
 
         render(500)
         small = render(2000)
         large = render(8000)
 
-        self.assertLess(large, 1.0)
-        self.assertLess(large, small * 8)
+        self.assertLess(large, small * 8 + 0.02)
 
     def test_dense_emphasis_is_near_linear(self):
         md = create_markdown()
@@ -33,7 +32,6 @@ def render(size):
         small = render(4000)
         large = render(8000)
 
-        self.assertLess(large, 1.0)
         self.assertLess(large, small * 3.5 + 0.02)
 
     def test_dense_emphasis_preserves_adjacent_tokens(self):

tests/test_security_formatting.py

@@ -1,13 +1,9 @@
-import platform
 import time
 from unittest import TestCase
 
 from mistune import create_markdown
 
 
-IS_PYPY = platform.python_implementation() == "PyPy"
-
-
 class TestFormattingSecurity(TestCase):
     def test_repeated_formatting_pairs_render_as_adjacent_spans(self):
         md = create_markdown(plugins=["strikethrough"])
@@ -25,8 +21,8 @@ def test_repeated_formatting_pairs_return_quickly(self):
                 md = create_markdown(plugins=[plugin])
                 md(unit)
 
-                repetitions = 500 if IS_PYPY else 3000
-                deadline = 1.0 if IS_PYPY else 0.5
+                repetitions = 3000
+                deadline = 1.0
                 text = unit * repetitions
 
                 start = time.monotonic()
@@ -46,13 +42,12 @@ def test_repeated_unclosed_formatting_markers_are_near_linear(self):
                 md = create_markdown(plugins=[plugin])
 
                 def render(size):
-                    start = time.process_time()
+                    start = time.perf_counter()
                     md(unit * size + "\n")
-                    return time.process_time() - start
+                    return time.perf_counter() - start
 
                 render(200)
                 small = render(1000)
                 large = render(2000)
 
-                self.assertLess(large, 1.0 if IS_PYPY else 0.5)
-                self.assertLess(large, small * 3)
+                self.assertLess(large, small * 3.5 + 0.02)

tests/test_security_inline.py

@@ -1,11 +1,9 @@
-import platform
 import time
 from unittest import TestCase
 
 from mistune import create_markdown
 
-IS_PYPY = platform.python_implementation() == "PyPy"
-DEADLINE = 1.0 if IS_PYPY else 0.5
+DEADLINE = 1.0
 
 
 class TestInlineParserSecurity(TestCase):
@@ -119,32 +117,30 @@ def test_repeated_link_suffixes_are_near_linear(self):
 
         def render(size):
             text = "[" * size + "a" + "](/u)" * size + "\n"
-            start = time.process_time()
+            start = time.perf_counter()
             md(text)
-            return time.process_time() - start
+            return time.perf_counter() - start
 
         render(200)
         small = render(1000)
         large = render(2000)
 
-        self.assertLess(large, DEADLINE)
-        self.assertLess(large, small * 3)
+        self.assertLess(large, small * 3.5 + 0.02)
 
     def test_repeated_unclosed_inline_spoilers_are_near_linear(self):
         md = create_markdown(plugins=["spoiler"])
 
         def render(size):
             text = ">! a " * size + "\n"
-            start = time.process_time()
+            start = time.perf_counter()
             md(text)
-            return time.process_time() - start
+            return time.perf_counter() - start
 
         render(200)
         small = render(1000)
         large = render(2000)
 
-        self.assertLess(large, DEADLINE)
-        self.assertLess(large, small * 3)
+        self.assertLess(large, small * 3.5 + 0.02)
 
     def test_adjacent_ruby_tokens_do_not_recurse_forever(self):
         text = "[a(b)]" * 3000 + "\n"

tests/test_security_math.py

@@ -1,12 +1,8 @@
-import platform
 import time
 from unittest import TestCase
 
 from mistune import create_markdown
 
-IS_PYPY = platform.python_implementation() == "PyPy"
-
-
 class TestMathSecurity(TestCase):
     def test_math_plugin_escapes_math_content(self):
         md = create_markdown(escape=True, plugins=["math"])
@@ -23,13 +19,12 @@ def test_repeated_inline_math_openers_are_near_linear(self):
         md = create_markdown(plugins=["math"])
 
         def render(size):
-            start = time.process_time()
+            start = time.perf_counter()
             md("$a " * size + "\n")
-            return time.process_time() - start
+            return time.perf_counter() - start
 
         render(200)
         small = render(1000)
         large = render(2000)
 
-        self.assertLess(large, 1.0 if IS_PYPY else 0.5)
-        self.assertLess(large, small * 3)
+        self.assertLess(large, small * 3.5 + 0.02)

tests/test_security_ref_links.py

@@ -1,16 +1,12 @@
-import platform
 import time
 from unittest import TestCase
 
 from mistune import create_markdown
 
-IS_PYPY = platform.python_implementation() == "PyPy"
-
-
 class TestRefLinkSecurity(TestCase):
     def test_many_reference_links_return_quickly(self):
-        repetitions = 500 if IS_PYPY else 3000
-        deadline = 1.0 if IS_PYPY else 0.5
+        repetitions = 3000
+        deadline = 1.0
 
         text = "".join("[a{}]: u\n".format(i) for i in range(repetitions))