1 files changed, 10 insertions, 11 deletions
diff --git a/tests/test_using_api.py b/tests/test_using_api.py
index 35546508..616bff6f 100644
--- a/tests/test_using_api.py
+++ b/tests/test_using_api.py
@@ -7,7 +7,7 @@
     :license: BSD, see LICENSE for details.
 """
 
-import unittest
+from pytest import raises
 
 from pygments.lexer import using, bygroups, this, RegexLexer
 from pygments.token import String, Text, Keyword
@@ -28,14 +28,13 @@ class MyLexer(RegexLexer):
     }
 
 
-class UsingStateTest(unittest.TestCase):
-    def test_basic(self):
-        expected = [(Text, 'a'), (String, '"'), (Keyword, 'bcd'),
-                    (String, '"'), (Text, 'e\n')]
-        t = list(MyLexer().get_tokens('a"bcd"e'))
-        self.assertEqual(t, expected)
+def test_basic():
+    expected = [(Text, 'a'), (String, '"'), (Keyword, 'bcd'),
+                (String, '"'), (Text, 'e\n')]
+    assert list(MyLexer().get_tokens('a"bcd"e')) == expected
 
-    def test_error(self):
-        def gen():
-            return list(MyLexer().get_tokens('#a'))
-        self.assertRaises(KeyError, gen)
+
+def test_error():
+    def gen():
+        return list(MyLexer().get_tokens('#a'))
+    assert raises(KeyError, gen)