From dbe2ef566d6fea7756565c05cf97e0cf035b6e7c Mon Sep 17 00:00:00 2001 From: Jorge Alvarez Jarreta Date: Thu, 19 Sep 2024 16:42:09 +0100 Subject: [PATCH] add new type alias for db_factory fixture --- src/ensembl/utils/plugin.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/ensembl/utils/plugin.py b/src/ensembl/utils/plugin.py index 5d8b9b9..a7a3334 100644 --- a/src/ensembl/utils/plugin.py +++ b/src/ensembl/utils/plugin.py @@ -20,7 +20,7 @@ import os from pathlib import Path import re -from typing import Callable, Generator +from typing import Callable, Generator, TypeAlias import pytest from pytest import Config, FixtureRequest, Parser @@ -30,6 +30,9 @@ from ensembl.utils.database import UnitTestDB +DBFactory: TypeAlias = Callable[[StrPath | None, str | None, MetaData | None], UnitTestDB] + + def pytest_addoption(parser: Parser) -> None: """Registers argparse-style options for Ensembl's unit testing. @@ -115,7 +118,7 @@ def _assert_files(result_path: StrPath, expected_path: StrPath) -> None: @pytest.fixture(name="db_factory", scope="module") -def fixture_db_factory(request: FixtureRequest, data_dir: Path) -> Generator[Callable, None, None]: +def fixture_db_factory(request: FixtureRequest, data_dir: Path) -> Generator[DBFactory, None, None]: """Yields a unit test database factory. Args: