scripts: mypy fixes

This commit is contained in:
Daniel Eklöf 2025-12-26 13:13:01 +01:00
parent cb1e152d99
commit bbebe0f330
No known key found for this signature in database
GPG key ID: 5BBD4992C116573F
5 changed files with 77 additions and 51 deletions

View file

@ -11,7 +11,7 @@ import termios
from datetime import datetime
def main():
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument('files', type=argparse.FileType('rb'), nargs='+')
parser.add_argument('--iterations', type=int, default=20)
@ -24,12 +24,12 @@ def main():
termios.TIOCGWINSZ,
struct.pack('HHHH', 0, 0, 0, 0)))
times = {name: [] for name in [f.name for f in args.files]}
times: dict[str, list[float]] = {name: [] for name in [f.name for f in args.files]}
for f in args.files:
bench_bytes = f.read()
for i in range(args.iterations):
for _ in range(args.iterations):
start = datetime.now()
sys.stdout.buffer.write(bench_bytes)
stop = datetime.now()
@ -48,4 +48,4 @@ def main():
if __name__ == '__main__':
sys.exit(main())
main()