You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
675 B
28 lines
675 B
7 months ago
|
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()
|