From 349bf9f0530f910ebdc2a8cbcc80b44957130deb Mon Sep 17 00:00:00 2001 From: Daniel Date: Thu, 7 Dec 2023 22:10:39 +0100 Subject: [PATCH] add stuff to __all__, simplify imports --- pytest_abra/__init__.py | 8 ++++++++ pytest_abra/cli.py | 2 +- recipes/authentik/tests_authentik/runner_authentik.py | 2 +- recipes/nextcloud/tests_nextcloud/runner_nextcloud.py | 2 +- recipes/wordpress/tests_wordpress/runner_wordpress.py | 2 +- 5 files changed, 12 insertions(+), 4 deletions(-) diff --git a/pytest_abra/__init__.py b/pytest_abra/__init__.py index e69de29..ebb3453 100644 --- a/pytest_abra/__init__.py +++ b/pytest_abra/__init__.py @@ -0,0 +1,8 @@ +from pytest_abra.coordinator import Coordinator +from pytest_abra.runner import Runner, Test + +__all__ = [ + "Coordinator", + "Runner", + "Test", +] diff --git a/pytest_abra/cli.py b/pytest_abra/cli.py index 8658dbb..30cd187 100644 --- a/pytest_abra/cli.py +++ b/pytest_abra/cli.py @@ -4,7 +4,7 @@ from pathlib import Path from loguru import logger -from pytest_abra.coordinator import Coordinator +from pytest_abra import Coordinator from pytest_abra.dir_manager import DirManager from pytest_abra.utils import get_datetime_string diff --git a/recipes/authentik/tests_authentik/runner_authentik.py b/recipes/authentik/tests_authentik/runner_authentik.py index b929945..10b9820 100644 --- a/recipes/authentik/tests_authentik/runner_authentik.py +++ b/recipes/authentik/tests_authentik/runner_authentik.py @@ -1,4 +1,4 @@ -from pytest_abra.runner import Runner, Test +from pytest_abra import Runner, Test def condition_always_true(env_config: dict[str, str]) -> bool: diff --git a/recipes/nextcloud/tests_nextcloud/runner_nextcloud.py b/recipes/nextcloud/tests_nextcloud/runner_nextcloud.py index f14e191..584239a 100644 --- a/recipes/nextcloud/tests_nextcloud/runner_nextcloud.py +++ b/recipes/nextcloud/tests_nextcloud/runner_nextcloud.py @@ -1,4 +1,4 @@ -from pytest_abra.runner import Runner, Test +from pytest_abra import Runner, Test def condition_always_false(env_config: dict[str, str]) -> bool: diff --git a/recipes/wordpress/tests_wordpress/runner_wordpress.py b/recipes/wordpress/tests_wordpress/runner_wordpress.py index e235809..0b5cb12 100644 --- a/recipes/wordpress/tests_wordpress/runner_wordpress.py +++ b/recipes/wordpress/tests_wordpress/runner_wordpress.py @@ -1,4 +1,4 @@ -from pytest_abra.runner import Runner, Test +from pytest_abra import Runner, Test def condition_always_true(env_config: dict[str, str]) -> bool: