summaryrefslogtreecommitdiff
path: root/pyflakes/test/test_match.py
blob: 89826e3115e74ecd24ae57faaec02f9bf386a30e (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
from sys import version_info

from pyflakes.test.harness import TestCase, skipIf


@skipIf(version_info < (3, 10), "Python >= 3.10 only")
class TestMatch(TestCase):
    def test_match_bindings(self):
        self.flakes('''
            def f():
                x = 1
                match x:
                    case 1 as y:
                        print(f'matched as {y}')
        ''')
        self.flakes('''
            def f():
                x = [1, 2, 3]
                match x:
                    case [1, y, 3]:
                        print(f'matched {y}')
        ''')
        self.flakes('''
            def f():
                x = {'foo': 1}
                match x:
                    case {'foo': y}:
                        print(f'matched {y}')
        ''')

    def test_match_pattern_matched_class(self):
        self.flakes('''
            from a import B

            match 1:
                case B(x=1) as y:
                    print(f'matched {y}')
        ''')
        self.flakes('''
            from a import B

            match 1:
                case B(a, x=z) as y:
                    print(f'matched {y} {a} {z}')
        ''')

    def test_match_placeholder(self):
        self.flakes('''
            def f():
                match 1:
                    case _:
                        print('catchall!')
        ''')

    def test_match_singleton(self):
        self.flakes('''
            match 1:
                case True:
                    print('true')
        ''')

    def test_match_or_pattern(self):
        self.flakes('''
            match 1:
                case 1 | 2:
                    print('one or two')
        ''')

    def test_match_star(self):
        self.flakes('''
            x = [1, 2, 3]
            match x:
                case [1, *y]:
                    print(f'captured: {y}')
        ''')

    def test_match_double_star(self):
        self.flakes('''
            x = {'foo': 'bar', 'baz': 'womp'}
            match x:
                case {'foo': k1, **rest}:
                    print(f'{k1=} {rest=}')
        ''')