Skip to content

Commit 81174d7

Browse files
committed
Add benchmark test suite for model operations
1 parent a9792e7 commit 81174d7

4 files changed

Lines changed: 650 additions & 0 deletions

File tree

make_sync.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,14 @@ def main():
5454
additional_replacements=ADDITIONAL_REPLACEMENTS,
5555
),
5656
]
57+
# Files to exclude from sync generation (benchmarks require special async handling)
58+
excluded_files = {"test_benchmarks.py"}
59+
5760
filepaths = []
5861
for root, _, filenames in os.walk(base_dir):
5962
for filename in filenames:
63+
if filename in excluded_files:
64+
continue
6065
if filename.rpartition(".")[-1] in (
6166
"py",
6267
"pyi",
@@ -107,5 +112,7 @@ def remove_run_async_call(match):
107112
f.write(content)
108113

109114

115+
116+
110117
if __name__ == "__main__":
111118
main()

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ dev = [
7272
"pre-commit>=4.3.0; python_version >= '3.10'",
7373
"mkdocs>=1.6.1",
7474
"mkdocs-material>=9.7.1",
75+
"pytest-benchmark>=5.2.3",
7576
]
7677

7778
[build-system]

0 commit comments

Comments
 (0)