summaryrefslogtreecommitdiff
blob: 2abaab19aab5ce673f2d9481df955082c5a1f562 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
https://github.com/jd-boyd/python-lzo/pull/66

diff --git a/setup.py b/setup.py
index 033692d..e7e383d 100644
--- a/setup.py
+++ b/setup.py
@@ -33,7 +33,7 @@ class TestCommand(Command):
         raise SystemExit(
             subprocess.call([sys.executable,
                              '-m',
-                             'nose']))
+                             'pytest']))
 
 
 if sys.platform == "win32":
@@ -91,7 +91,7 @@ setup_args = get_kw(
     maintainer_email="jdboyd@jdboyd.net",
     url="https://github.com/jd-boyd/python-lzo",
     license="GNU General Public License (GPL)",
-    tests_require=['nose'],
+    tests_require=['pytest'],
     cmdclass={
         'test': TestCommand
     },
diff --git a/tests/test.py b/tests/test_lzo.py
similarity index 88%
rename from tests/test.py
rename to tests/test_lzo.py
index af761d9..b9d5c88 100644
--- a/tests/test.py
+++ b/tests/test_lzo.py
@@ -32,6 +32,7 @@
 
 from __future__ import print_function
 
+import pytest
 import sys, string
 
 # update sys.path when running in the build directory
@@ -94,22 +95,20 @@ def test_version():
     assert pkg_version == mod_version, \
         "%r != %r" %(pkg_version, mod_version)
 
-def test_lzo():
-    yield gen, b"aaaaaaaaaaaaaaaaaaaaaaaa"
-    yield gen, b"abcabcabcabcabcabcabcabc"
-    yield gen, b"abcabcabcabcabcabcabcabc", 9
+@pytest.mark.parametrize("src, level", [(b"aaaaaaaaaaaaaaaaaaaaaaaa", 1), (b"abcabcabcabcabcabcabcabc", 1), (b"abcabcabcabcabcabcabcabc", 9)])
+def test_lzo(src, level):
+    gen(src, level)
 
-
-def test_lzo_raw():
-    yield gen_raw, b"aaaaaaaaaaaaaaaaaaaaaaaa"
-    yield gen_raw, b"abcabcabcabcabcabcabcabc"
-    yield gen_raw, b"abcabcabcabcabcabcabcabc", 9
+@pytest.mark.parametrize("src, level", [(b"aaaaaaaaaaaaaaaaaaaaaaaa", 1), (b"abcabcabcabcabcabcabcabc", 1), (b"abcabcabcabcabcabcabcabc", 9)])
+def test_lzo_raw(src, level):
+    gen_raw(src, level)
 
 
 def test_lzo_empty():
-    yield gen, b""
-    yield gen_raw, b""
+    gen(b"")
 
+def test_lzo_empty_raw():
+    gen_raw(b"")
 
 def test_lzo_big():
     gen(b" " * 131072)