| try: |
| # python 2.x |
| import unittest2 as unittest |
| except ImportError: |
| # python 3.x |
| import unittest |
| |
| import funcsigs |
| |
| |
| class TestFormatAnnotation(unittest.TestCase): |
| def test_string (self): |
| self.assertEqual(funcsigs.formatannotation("annotation"), |
| "'annotation'") |
| |
| def test_builtin_type (self): |
| self.assertEqual(funcsigs.formatannotation(int), |
| "int") |
| |
| def test_user_type (self): |
| class dummy (object): pass |
| self.assertEqual(funcsigs.formatannotation(dummy), |
| "tests.test_formatannotation.dummy") |
| |
| |
| if __name__ == "__main__": |
| unittest.begin() |