1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
#!/usr/bin/env python3
# Copyright (c) Facebook, Inc. and its affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.
#
"""
############################################################################
# Python 3 only tests. This is not expected to work on python 2.
# Any test that should work on both python 2 and 3 should go to tests.py
############################################################################
"""
import typing
import unittest
from nubia.internal.typing import argument, inspect_object
class DecoratorTest(unittest.TestCase):
def test_equality_decorated(self):
@argument("arg1", description="arg1 desc")
@argument("arg2", description="arg2 desc")
def foo(arg1: typing.Any, arg2: str) -> typing.Tuple[str, str]:
return (arg1, arg2)
@argument("arg1", type=typing.Any, description="arg1 desc")
@argument("arg2", type=str, description="arg2 desc")
def bar(arg1, arg2):
return (arg1, arg2)
self.assertEqual(inspect_object(foo), inspect_object(bar))
def test_inequality_no_decorator(self):
def foo(arg1: str, arg2: str) -> typing.Tuple[str, str]:
return (arg1, arg2)
def bar(arg1, arg2):
return (arg1, arg2)
self.assertNotEqual(inspect_object(foo), inspect_object(bar))
def test_inequality_decorated(self):
def foo(arg1: str, arg2: str) -> typing.Tuple[str, str]:
return (arg1, arg2)
@argument("arg1", type=int)
@argument("arg2", type=int)
def bar(arg1, arg2):
return (arg1, arg2)
self.assertNotEqual(inspect_object(foo), inspect_object(bar))
def test_type_conflict(self):
# specifiying arg as str in both the decorator and in the type
# annotation is redundant but should be fine
@argument("arg", type=str)
def foo(arg: str) -> str:
return arg
try:
# arg is being specified as str by the decorator but as typing.Any
# by the type annotation. A TypeError should be raised
@argument("arg", type=str)
def bar(arg: typing.Any) -> str:
return arg
self.fail(
"foo declaration should fail with TypeError as it "
"declares arg as both str and typing.Any"
)
except TypeError:
pass
|
