From ef5334ba3df0c4151c8f3b1f9294b3cfa05c3837 Mon Sep 17 00:00:00 2001 From: tazlin Date: Mon, 3 Jun 2024 15:20:14 -0400 Subject: [PATCH] test: tweaks for consistency --- tests/conftest.py | 2 +- tests/test_image.py | 1 + tests/test_image_extra_sources.py | 1 + 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index c449802d..6c39bf77 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -23,7 +23,7 @@ def api_key() -> str: raise ValueError("No api key file found") -@pytest.fixture(autouse=True) +@pytest.fixture(autouse=True, scope="session") def increase_kudos(api_key: str, HORDE_URL: str, CIVERSION: str) -> None: headers = {"apikey": api_key, "Client-Agent": f"aihorde_ci_client:{CIVERSION}:(discord)db0#1625", "user_id": "1"} diff --git a/tests/test_image.py b/tests/test_image.py index 4bf7a704..a464d155 100644 --- a/tests/test_image.py +++ b/tests/test_image.py @@ -38,6 +38,7 @@ def test_simple_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None: "name": "CICD Fake Dreamer", "models": TEST_MODELS, "bridge_agent": "AI Horde Worker reGen:8.0.1-citests:https://github.com/Haidra-Org/horde-worker-reGen", + "nsfw": True, "amount": 10, "max_pixels": 4194304, "allow_img2img": True, diff --git a/tests/test_image_extra_sources.py b/tests/test_image_extra_sources.py index 1d05c7a2..24e8bc49 100644 --- a/tests/test_image_extra_sources.py +++ b/tests/test_image_extra_sources.py @@ -56,6 +56,7 @@ def test_simple_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None: "name": "CICD Fake Dreamer", "models": TEST_MODELS, "bridge_agent": "AI Horde Worker reGen:8.0.1-citests:https://github.com/Haidra-Org/horde-worker-reGen", + "nsfw": True, "amount": 10, "max_pixels": 4194304, "allow_img2img": True,