justin/OpenVixDiskLib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix code formatting - tox -e fmt

Browse Source
This commit is contained in:
Lucian Petrut
2026-09-08 14:59:43 +00:00
parent ffce3a1cd6
commit e595dcd9e7
13 changed files with 615 additions and 492 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/perf/test_compare.py
+45 -31
View File
@@ -10,7 +10,7 @@ from typing import Any
from openvixdisklib import openvixdisklib as open_vix
from tests.integration import vixdisklib
from tests.integration.base import LabEnv, SECTOR_SIZE, pattern_bytes
from tests.integration.base import SECTOR_SIZE, LabEnv, pattern_bytes
_SIZES = (
("64KiB", 64 * 1024),
@@ -19,8 +19,7 @@ _SIZES = (
)
def _connect_extra(
lab: LabEnv, module: Any, transport_mode: str) -> dict[str, Any]:
def _connect_extra(lab: LabEnv, module: Any, transport_mode: str) -> dict[str, Any]:
"""Return extra ``connect`` kwargs needed by ``module``."""
extra: dict[str, Any] = {"transport_modes": transport_mode}
if module is open_vix:
@@ -29,31 +28,31 @@ def _connect_extra(
def _time_write_read(
lab: LabEnv,
module: Any,
payload: bytes,
flags: int = 0,
transport_mode: str = "nbdssl") -> tuple[float, float]:
lab: LabEnv,
module: Any,
payload: bytes,
flags: int = 0,
transport_mode: str = "nbdssl",
) -> tuple[float, float]:
"""Write ``payload`` at sector 0, read it back, and return durations."""
n_sectors = len(payload) // SECTOR_SIZE
handle = module.VixDiskLibHandle(
vixdisklib_compatibility_version="8.0",
config_path=None)
vixdisklib_compatibility_version="8.0", config_path=None
)
write_buf = module.get_buffer(len(payload))
read_buf = module.get_buffer(len(payload))
write_buf[:len(payload)] = payload
kwargs = lab.vixdisklib_connect_kwargs(
_connect_extra(lab, module, transport_mode))
write_buf[: len(payload)] = payload
kwargs = lab.vixdisklib_connect_kwargs(_connect_extra(lab, module, transport_mode))
with handle.connect(**kwargs) as conn:
with handle.open(conn, lab.disk_path, flags=flags) as disk:
started = time.perf_counter()
handle.write(disk, 0, n_sectors, write_buf)
write_s = time.perf_counter() - started
read_buf[:len(payload)] = b"\xa5" * len(payload)
read_buf[: len(payload)] = b"\xa5" * len(payload)
started = time.perf_counter()
handle.read(disk, 0, n_sectors, read_buf)
read_s = time.perf_counter() - started
assert read_buf.raw[:len(payload)] == payload
assert read_buf.raw[: len(payload)] == payload
return write_s, read_s
@@ -82,27 +81,42 @@ class TestCompare:
for mode_name, flags in open_modes:
for name, module in libraries:
write_s, read_s = _time_write_read(
lab, module, payload, flags=flags,
transport_mode=transport_mode)
rows.append((
label,
transport_mode,
mode_name,
name,
write_s,
read_s,
_mib_per_s(nbytes, write_s),
_mib_per_s(nbytes, read_s),
))
lab,
module,
payload,
flags=flags,
transport_mode=transport_mode,
)
rows.append(
(
label,
transport_mode,
mode_name,
name,
write_s,
read_s,
_mib_per_s(nbytes, write_s),
_mib_per_s(nbytes, read_s),
)
)
print()
print(
f"{'size':<14} {'transport':<10} {'flags':<8} {'library':<16} "
f"{'write_s':>10} {'read_s':>10} "
f"{'write_MiB/s':>12} {'read_MiB/s':>12}")
f"{'write_MiB/s':>12} {'read_MiB/s':>12}"
)
for (
label, transport_mode, mode_name, name,
write_s, read_s, write_r, read_r) in rows:
label,
transport_mode,
mode_name,
name,
write_s,
read_s,
write_r,
read_r,
) in rows:
print(
f"{label:<14} {transport_mode:<10} {mode_name:<8} {name:<16} "
f"{write_s:10.3f} {read_s:10.3f} "
f"{write_r:12.1f} {read_r:12.1f}")
f"{write_r:12.1f} {read_r:12.1f}"
)