28 lines
675 B
Python
28 lines
675 B
Python
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()
|