import funcsigs import unittest2 as unittest 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")