stop using unittestpytest

1 files changed, 29 insertions, 0 deletions

diff --git a/tests/test_leak.py b/tests/test_leak.py
new file mode 100644
index 0000000..964341a
--- /dev/null
+++ b/tests/test_leak.py
@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+import gc
+import sys
+
+import pytest
+
+from markupsafe import escape
+
+
+@pytest.mark.skipif(
+    hasattr(escape, 'func_code'),
+    reason='only test memory leak with speedups'
+)
+def test_markup_leaks():
+    counts = set()
+
+    for count in range(20):
+        for item in range(1000):
+            escape("foo")
+            escape("<foo>")
+            escape(u"foo")
+            escape(u"<foo>")
+
+        if hasattr(sys, 'pypy_version_info'):
+            gc.collect()
+
+        counts.add(len(gc.get_objects()))
+
+    assert len(counts) == 1