tests: faster tests

1 year ago · 6db296aa37
parent 6f9e9f64a9
commit 6db296aa37
35 changed files with 6 additions and 2 deletions
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0.05]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0.05]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0.2]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0.2]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0.05]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0.05]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0.2]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0.2]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0.05]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0.05]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0.2]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0.2]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0.05]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0.05]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0.2]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0.2]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0.05]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0.05]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0.2]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0.2]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0.05]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0.05]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0.2]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0.2]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0.05]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0.05]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0.2]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0.2]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0.05]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0.05]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0.2]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0.2]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.05]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.05]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.2]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.2]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0.05]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0.05]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0.2]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0.2]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0.05]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0.05]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0.2]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0.2]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0.05]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0.05]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0.2]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0.2]_.png
--- a/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0]_.png
+++ b/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0]_.png
--- a/tests/test_api.py
+++ b/tests/test_api.py
@ -173,7 +173,11 @@ def test_img_to_img_fruit_2_gold(
    img = LazyLoadingImage(
        filepath=os.path.join(TESTS_FOLDER, "data", "bowl_of_fruit.jpg")
    )
-
+    target_steps = 25
+    if init_strength >= 1:
+        needed_steps = 25
+    else:
+        needed_steps = int(target_steps / (1 - init_strength))
    prompt = ImaginePrompt(
        "a white bowl filled with gold coins",
        prompt_strength=12,
@ -181,7 +185,7 @@ def test_img_to_img_fruit_2_gold(
        init_image_strength=init_strength,
        mask_prompt="(fruit{*2} OR stem{*10} OR fruit stem{*3})",
        mask_mode="replace",
-        steps=80,
+        steps=needed_steps,
        seed=1,
        sampler_type=sampler_type,
    )