mirror of
https://github.com/capstone-engine/llvm-capstone.git
synced 2025-05-18 03:46:39 +00:00

This is the first commit in a series that will reformat all the python files in the LLVM repository. Reformatting is done with `black`. See more information here: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Reviewed By: jhenderson, JDevlieghere, MatzeB Differential Revision: https://reviews.llvm.org/D150545
58 lines
1.3 KiB
Python
58 lines
1.3 KiB
Python
#!/usr/bin/env python
|
|
|
|
import os
|
|
import sys
|
|
|
|
if len(sys.argv) == 3 and sys.argv[1] == "--gtest_list_tests":
|
|
if sys.argv[2] != "--gtest_filter=-*DISABLED_*":
|
|
raise ValueError("unexpected argument: %s" % (sys.argv[2]))
|
|
print(
|
|
"""\
|
|
FirstTest.
|
|
subTestA"""
|
|
)
|
|
sys.exit(0)
|
|
elif len(sys.argv) != 1:
|
|
# sharding and json output are specified using environment variables
|
|
raise ValueError("unexpected argument: %r" % (" ".join(sys.argv[1:])))
|
|
|
|
for e in ["GTEST_TOTAL_SHARDS", "GTEST_SHARD_INDEX", "GTEST_OUTPUT"]:
|
|
if e not in os.environ:
|
|
raise ValueError("missing environment variables: " + e)
|
|
|
|
if not os.environ["GTEST_OUTPUT"].startswith("json:"):
|
|
raise ValueError("must emit json output: " + os.environ["GTEST_OUTPUT"])
|
|
|
|
output = """\
|
|
{
|
|
"random_seed": 123,
|
|
"testsuites": [
|
|
{
|
|
"name": "FirstTest",
|
|
"testsuite": [
|
|
{
|
|
"name": "subTestA",
|
|
"result": "COMPLETED",
|
|
"time": "0.001s"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}"""
|
|
|
|
dummy_output = """\
|
|
{
|
|
"testsuites": [
|
|
]
|
|
}"""
|
|
|
|
json_filename = os.environ["GTEST_OUTPUT"].split(":", 1)[1]
|
|
with open(json_filename, "w", encoding="utf-8") as f:
|
|
if os.environ["GTEST_SHARD_INDEX"] == "0":
|
|
f.write(output)
|
|
else:
|
|
f.write(dummy_output)
|
|
exit_code = 0
|
|
|
|
sys.exit(exit_code)
|