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
|
# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
from typing import Any
import libcst as cst
from libcst._nodes.tests.base import CSTNodeTest
from libcst.metadata import CodeRange
from libcst.testing.utils import data_provider
class SmallStatementTest(CSTNodeTest):
@data_provider(
(
{"node": cst.Pass(), "code": "pass"},
{"node": cst.Pass(semicolon=cst.Semicolon()), "code": "pass;"},
{
"node": cst.Pass(
semicolon=cst.Semicolon(
whitespace_before=cst.SimpleWhitespace(" "),
whitespace_after=cst.SimpleWhitespace(" "),
)
),
"code": "pass ; ",
"expected_position": CodeRange((1, 0), (1, 4)),
},
{"node": cst.Continue(), "code": "continue"},
{"node": cst.Continue(semicolon=cst.Semicolon()), "code": "continue;"},
{
"node": cst.Continue(
semicolon=cst.Semicolon(
whitespace_before=cst.SimpleWhitespace(" "),
whitespace_after=cst.SimpleWhitespace(" "),
)
),
"code": "continue ; ",
"expected_position": CodeRange((1, 0), (1, 8)),
},
{"node": cst.Break(), "code": "break"},
{"node": cst.Break(semicolon=cst.Semicolon()), "code": "break;"},
{
"node": cst.Break(
semicolon=cst.Semicolon(
whitespace_before=cst.SimpleWhitespace(" "),
whitespace_after=cst.SimpleWhitespace(" "),
)
),
"code": "break ; ",
"expected_position": CodeRange((1, 0), (1, 5)),
},
{
"node": cst.Expr(
cst.BinaryOperation(cst.Name("x"), cst.Add(), cst.Name("y"))
),
"code": "x + y",
},
{
"node": cst.Expr(
cst.BinaryOperation(cst.Name("x"), cst.Add(), cst.Name("y")),
semicolon=cst.Semicolon(),
),
"code": "x + y;",
},
{
"node": cst.Expr(
cst.BinaryOperation(cst.Name("x"), cst.Add(), cst.Name("y")),
semicolon=cst.Semicolon(
whitespace_before=cst.SimpleWhitespace(" "),
whitespace_after=cst.SimpleWhitespace(" "),
),
),
"code": "x + y ; ",
"expected_position": CodeRange((1, 0), (1, 5)),
},
)
)
def test_valid(self, **kwargs: Any) -> None:
self.validate_node(**kwargs)
|
