mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
d8c8ba2440
Pull Request resolved: https://github.com/pytorch/pytorch/pull/136964 Approved by: https://github.com/justinchuby, https://github.com/albanD
54 lines
1.3 KiB
Python
54 lines
1.3 KiB
Python
# 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, _ = 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()
|