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
|
# Owner(s): ["module: unknown"]
from torch.testing import FileCheck
from torch.testing._internal.common_utils import run_tests, TestCase
class TestFileCheck(TestCase):
def test_not_run(self):
stdout, stderr = self.run_process_no_exception(
"""\
from torch.testing import FileCheck
file_check = FileCheck().check("not run")
del file_check
""",
)
FileCheck().check("You have not run this instance of FileCheck!").check_next(
"FileCheck checks:"
).check_next("\tCHECK: not run").run(stdout)
def test_all_python_api(self):
test_string = """
check check_same
check_next
check_count
check_dag
check_source_highlighted
~~~~~~~~~~~~~~~~~~~~~~~~
check_regex
"""
FileCheck().check("check").check_not("check_not").check_same(
"check_same"
).check_next("check_next").check_count("check_count", 1).check_dag(
"check_dag"
).check_source_highlighted("check_source_highlighted").check_regex(
r"check_.+"
).run(test_string)
FileCheck().run(
"""
# CHECK: check
# CHECK-NOT: check_not
# CHECK-SAME: check_same
# CHECK-NEXT: check_next
# CHECK-DAG: check_dag
# CHECK-SOURCE-HIGHLIGHTED: check_source_highlighted
# CHECK-REGEX: check_.+
""",
test_string,
)
if __name__ == "__main__":
run_tests()
|
