diff --git a/web3/scripts/release/test_package.py b/web3/scripts/release/test_package.py index bf75f745d3..cd3d6dc6ee 100644 --- a/web3/scripts/release/test_package.py +++ b/web3/scripts/release/test_package.py @@ -5,17 +5,20 @@ from tempfile import ( TemporaryDirectory, ) +from typing import ( + Tuple, +) import venv -def create_venv(parent_path): +def create_venv(parent_path: Path) -> Path: venv_path = parent_path / 'package-smoke-test' venv.create(venv_path, with_pip=True) subprocess.run([venv_path / 'bin' / 'pip', 'install', '-U', 'pip', 'setuptools'], check=True) return venv_path -def find_wheel(project_path): +def find_wheel(project_path: Path) -> Path: wheels = list(project_path.glob('dist/*.whl')) if len(wheels) != 1: @@ -26,7 +29,9 @@ def find_wheel(project_path): return wheels[0] -def install_wheel(venv_path, wheel_path, extras=()): +def install_wheel( + venv_path: Path, wheel_path: Path, extras: Tuple[str, ...] = () +) -> None: if extras: extra_suffix = f"[{','.join(extras)}]" else: @@ -42,7 +47,7 @@ def install_wheel(venv_path, wheel_path, extras=()): ) -def test_install_local_wheel(): +def test_install_local_wheel() -> None: with TemporaryDirectory() as tmpdir: venv_path = create_venv(Path(tmpdir)) wheel_path = find_wheel(Path('.'))