Cython/Compiler/Tests/TestDecorators.py

   1 import unittest
   2 from Cython.TestUtils import TransformTest
   3 from Cython.Compiler.ParseTreeTransforms import DecoratorTransform
   4
   5 class TestDecorator(TransformTest):
   6
   7     def test_decorator(self):
   8         t = self.run_pipeline([DecoratorTransform(None)], u"""
   9         def decorator(fun):
  10             return fun
  11         @decorator
  12         def decorated():
  13             pass
  14         """)
  15
  16         self.assertCode(u"""
  17         def decorator(fun):
  18             return fun
  19         def decorated():
  20             pass
  21         decorated = decorator(decorated)
  22         """, t)
  23
  24 if __name__ == '__main__':
  25     unittest.main()