1 files changed, 21 insertions, 19 deletions
diff --git a/tests/test_cpp.py b/tests/test_cpp.py
index ef59965c..b87b2dd3 100644
--- a/tests/test_cpp.py
+++ b/tests/test_cpp.py
@@ -7,27 +7,29 @@
     :license: BSD, see LICENSE for details.
 """
 
-import unittest
+import pytest
 
 from pygments.lexers import CppLexer
 from pygments.token import Token
 
 
-class CppTest(unittest.TestCase):
-    def setUp(self):
-        self.lexer = CppLexer()
-
-    def testGoodComment(self):
-        fragment = u'/* foo */\n'
-        tokens = [
-            (Token.Comment.Multiline, u'/* foo */'),
-            (Token.Text, u'\n'),
-        ]
-        self.assertEqual(tokens, list(self.lexer.get_tokens(fragment)))
-
-    def testOpenComment(self):
-        fragment = u'/* foo\n'
-        tokens = [
-            (Token.Comment.Multiline, u'/* foo\n'),
-        ]
-        self.assertEqual(tokens, list(self.lexer.get_tokens(fragment)))
+@pytest.fixture(scope='module')
+def lexer():
+    yield CppLexer()
+
+
+def test_good_comment(lexer):
+    fragment = u'/* foo */\n'
+    tokens = [
+        (Token.Comment.Multiline, u'/* foo */'),
+        (Token.Text, u'\n'),
+    ]
+    assert list(lexer.get_tokens(fragment)) == tokens
+
+
+def test_open_comment(lexer):
+    fragment = u'/* foo\n'
+    tokens = [
+        (Token.Comment.Multiline, u'/* foo\n'),
+    ]
+    assert list(lexer.get_tokens(fragment)) == tokens