summaryrefslogtreecommitdiff
path: root/lib/python2.7/test/test_complex_args.py
blob: 036cbea4029b91e2428feaa208678b2a6c2ab99d (plain)
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

import unittest
from test import test_support
import textwrap

class ComplexArgsTestCase(unittest.TestCase):

    def check(self, func, expected, *args):
        self.assertEqual(func(*args), expected)

    # These functions are tested below as lambdas too.  If you add a
    # function test, also add a similar lambda test.

    # Functions are wrapped in "exec" statements in order to
    # silence Py3k warnings.

    def test_func_parens_no_unpacking(self):
        exec textwrap.dedent("""
        def f(((((x))))): return x
        self.check(f, 1, 1)
        # Inner parens are elided, same as: f(x,)
        def f(((x)),): return x
        self.check(f, 2, 2)
        """)

    def test_func_1(self):
        exec textwrap.dedent("""
        def f(((((x),)))): return x
        self.check(f, 3, (3,))
        def f(((((x)),))): return x
        self.check(f, 4, (4,))
        def f(((((x))),)): return x
        self.check(f, 5, (5,))
        def f(((x),)): return x
        self.check(f, 6, (6,))
        """)

    def test_func_2(self):
        exec textwrap.dedent("""
        def f(((((x)),),)): return x
        self.check(f, 2, ((2,),))
        """)

    def test_func_3(self):
        exec textwrap.dedent("""
        def f((((((x)),),),)): return x
        self.check(f, 3, (((3,),),))
        """)

    def test_func_complex(self):
        exec textwrap.dedent("""
        def f((((((x)),),),), a, b, c): return x, a, b, c
        self.check(f, (3, 9, 8, 7), (((3,),),), 9, 8, 7)

        def f(((((((x)),)),),), a, b, c): return x, a, b, c
        self.check(f, (3, 9, 8, 7), (((3,),),), 9, 8, 7)

        def f(a, b, c, ((((((x)),)),),)): return a, b, c, x
        self.check(f, (9, 8, 7, 3), 9, 8, 7, (((3,),),))
        """)

    # Duplicate the tests above, but for lambda.  If you add a lambda test,
    # also add a similar function test above.

    def test_lambda_parens_no_unpacking(self):
        exec textwrap.dedent("""
        f = lambda (((((x))))): x
        self.check(f, 1, 1)
        # Inner parens are elided, same as: f(x,)
        f = lambda ((x)),: x
        self.check(f, 2, 2)
        """)

    def test_lambda_1(self):
        exec textwrap.dedent("""
        f = lambda (((((x),)))): x
        self.check(f, 3, (3,))
        f = lambda (((((x)),))): x
        self.check(f, 4, (4,))
        f = lambda (((((x))),)): x
        self.check(f, 5, (5,))
        f = lambda (((x),)): x
        self.check(f, 6, (6,))
        """)

    def test_lambda_2(self):
        exec textwrap.dedent("""
        f = lambda (((((x)),),)): x
        self.check(f, 2, ((2,),))
        """)

    def test_lambda_3(self):
        exec textwrap.dedent("""
        f = lambda ((((((x)),),),)): x
        self.check(f, 3, (((3,),),))
        """)

    def test_lambda_complex(self):
        exec textwrap.dedent("""
        f = lambda (((((x)),),),), a, b, c: (x, a, b, c)
        self.check(f, (3, 9, 8, 7), (((3,),),), 9, 8, 7)

        f = lambda ((((((x)),)),),), a, b, c: (x, a, b, c)
        self.check(f, (3, 9, 8, 7), (((3,),),), 9, 8, 7)

        f = lambda a, b, c, ((((((x)),)),),): (a, b, c, x)
        self.check(f, (9, 8, 7, 3), 9, 8, 7, (((3,),),))
        """)


def test_main():
    with test_support.check_py3k_warnings(
            ("tuple parameter unpacking has been removed", SyntaxWarning),
            ("parenthesized argument names are invalid", SyntaxWarning)):
        test_support.run_unittest(ComplexArgsTestCase)

if __name__ == "__main__":
    test_main()