Minor cleanup in test

1 files changed, 8 insertions, 9 deletions

diff --git a/tests/test_c_generator.py b/tests/test_c_generator.py
index d024616..e5fd9db 100644
--- a/tests/test_c_generator.py
+++ b/tests/test_c_generator.py
@@ -35,23 +35,22 @@ def parse_to_ast(src):
     return _c_parser.parse(src)
 
 
-class FuncDeclVisitor(c_ast.NodeVisitor):
-    def __init__(self):
-        self.stubs = []
-
-    def visit_FuncDecl(self, node):
-        gen = c_generator.CGenerator()
-        self.stubs.append(gen.visit(node))
+class TestFunctionDeclGeneration(unittest.TestCase):
+    class _FuncDeclVisitor(c_ast.NodeVisitor):
+        def __init__(self):
+            self.stubs = []
 
+        def visit_FuncDecl(self, node):
+            gen = c_generator.CGenerator()
+            self.stubs.append(gen.visit(node))
 
-class TestFunctionDeclGeneration(unittest.TestCase):
     def test_partial_funcdecl_generation(self):
         src = r'''
             void noop(void);
             void *something(void *thing);
             int add(int x, int y);'''
         ast = parse_to_ast(src)
-        v = FuncDeclVisitor()
+        v = TestFunctionDeclGeneration._FuncDeclVisitor()
         v.visit(ast)
         self.assertEqual(len(v.stubs), 3)
         self.assertTrue(r'void noop(void)' in v.stubs)