[Refactor] Move testing utils out of src (#12238)

* update

* update

* update

* update

* update

* merge main

* Revert "merge main"

This reverts commit 65efbcead58644b31596ed2d714f7cee0e0238d3.

tests/models/unets/test_models_unet_spatiotemporal.py

@@ -21,13 +21,13 @@ import torch
 from diffusers import UNetSpatioTemporalConditionModel
 from diffusers.utils import logging
 from diffusers.utils.import_utils import is_xformers_available
-from diffusers.utils.testing_utils import (
+
+from ...testing_utils import (
     enable_full_determinism,
     floats_tensor,
     skip_mps,
     torch_device,
 )
-
 from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
 
 

tests/models/unets/test_unet_2d_blocks.py

@@ -15,8 +15,8 @@
 import unittest
 
 from diffusers.models.unets.unet_2d_blocks import *  # noqa F403
-from diffusers.utils.testing_utils import torch_device
 
+from ...testing_utils import torch_device
 from .test_unet_blocks_common import UNetBlockTesterMixin
 
 

tests/models/unets/test_unet_blocks_common.py

@@ -16,14 +16,15 @@ from typing import Tuple
 
 import torch
 
-from diffusers.utils.testing_utils import (
+from diffusers.utils.torch_utils import randn_tensor
+
+from ...testing_utils import (
     floats_tensor,
     require_torch,
     require_torch_accelerator_with_training,
     torch_all_close,
     torch_device,
 )
-from diffusers.utils.torch_utils import randn_tensor
 
 
 @require_torch

tests/modular_pipelines/stable_diffusion_xl/test_modular_pipeline_stable_diffusion_xl.py

@@ -27,15 +27,15 @@ from diffusers import (
     StableDiffusionXLModularPipeline,
 )
 from diffusers.loaders import ModularIPAdapterMixin
-from diffusers.utils.testing_utils import (
-    enable_full_determinism,
-    floats_tensor,
-    torch_device,
-)
 
 from ...models.unets.test_models_unet_2d_condition import (
     create_ip_adapter_state_dict,
 )
+from ...testing_utils import (
+    enable_full_determinism,
+    floats_tensor,
+    torch_device,
+)
 from ..test_modular_pipelines_common import (
     ModularPipelineTesterMixin,
 )

tests/modular_pipelines/test_modular_pipelines_common.py

@@ -9,7 +9,8 @@ import torch
 import diffusers
 from diffusers import ComponentsManager, ModularPipeline, ModularPipelineBlocks
 from diffusers.utils import logging
-from diffusers.utils.testing_utils import (
+
+from ..testing_utils import (
     backend_empty_cache,
     numpy_cosine_similarity_distance,
     require_accelerator,

tests/others/test_config.py

@@ -28,7 +28,8 @@ from diffusers import (
     logging,
 )
 from diffusers.configuration_utils import ConfigMixin, register_to_config
-from diffusers.utils.testing_utils import CaptureLogger
+
+from ..testing_utils import CaptureLogger
 
 
 class SampleObject(ConfigMixin):

tests/others/test_ema.py

@@ -20,7 +20,8 @@ import torch
 
 from diffusers import UNet2DConditionModel
 from diffusers.training_utils import EMAModel
-from diffusers.utils.testing_utils import enable_full_determinism, skip_mps, torch_device
+
+from ..testing_utils import enable_full_determinism, skip_mps, torch_device
 
 
 enable_full_determinism()

tests/others/test_outputs.py

@@ -7,7 +7,8 @@ import numpy as np
 import PIL.Image
 
 from diffusers.utils.outputs import BaseOutput
-from diffusers.utils.testing_utils import require_torch
+
+from ..testing_utils import require_torch
 
 
 @dataclass

tests/others/test_training.py

@@ -19,7 +19,8 @@ import torch
 
 from diffusers import DDIMScheduler, DDPMScheduler, UNet2DModel
 from diffusers.training_utils import set_seed
-from diffusers.utils.testing_utils import slow
+
+from ..testing_utils import slow
 
 
 torch.backends.cuda.matmul.allow_tf32 = False

tests/others/test_utils.py

@@ -20,7 +20,8 @@ import pytest
 
 from diffusers import __version__
 from diffusers.utils import deprecate
-from diffusers.utils.testing_utils import Expectations, str_to_bool
+
+from ..testing_utils import Expectations, str_to_bool
 
 
 # Used to test the hub

tests/pipelines/allegro/test_allegro.py

@@ -23,7 +23,8 @@ import torch
 from transformers import AutoTokenizer, T5Config, T5EncoderModel
 
 from diffusers import AllegroPipeline, AllegroTransformer3DModel, AutoencoderKLAllegro, DDIMScheduler
-from diffusers.utils.testing_utils import (
+
+from ...testing_utils import (
     backend_empty_cache,
     enable_full_determinism,
     numpy_cosine_similarity_distance,
@@ -33,7 +34,6 @@ from diffusers.utils.testing_utils import (
     slow,
     torch_device,
 )
-
 from ..pipeline_params import TEXT_TO_IMAGE_BATCH_PARAMS, TEXT_TO_IMAGE_IMAGE_PARAMS, TEXT_TO_IMAGE_PARAMS
 from ..test_pipelines_common import PipelineTesterMixin, PyramidAttentionBroadcastTesterMixin, to_np
 

tests/pipelines/animatediff/test_animatediff.py

@@ -19,7 +19,8 @@ from diffusers import (
 )
 from diffusers.models.attention import FreeNoiseTransformerBlock
 from diffusers.utils import is_xformers_available, logging
-from diffusers.utils.testing_utils import (
+
+from ...testing_utils import (
     backend_empty_cache,
     numpy_cosine_similarity_distance,
     require_accelerator,
@@ -27,7 +28,6 @@ from diffusers.utils.testing_utils import (
     slow,
     torch_device,
 )
-
 from ..pipeline_params import TEXT_TO_IMAGE_BATCH_PARAMS, TEXT_TO_IMAGE_PARAMS
 from ..test_pipelines_common import (
     IPAdapterTesterMixin,

tests/pipelines/animatediff/test_animatediff_controlnet.py

@@ -21,8 +21,8 @@ from diffusers import (
 from diffusers.models.attention import FreeNoiseTransformerBlock
 from diffusers.utils import logging
 from diffusers.utils.import_utils import is_xformers_available
-from diffusers.utils.testing_utils import require_accelerator, torch_device
 
+from ...testing_utils import require_accelerator, torch_device
 from ..pipeline_params import TEXT_TO_IMAGE_BATCH_PARAMS, TEXT_TO_IMAGE_PARAMS
 from ..test_pipelines_common import (
     IPAdapterTesterMixin,

tests/pipelines/animatediff/test_animatediff_sdxl.py

@@ -14,8 +14,8 @@ from diffusers import (
     UNetMotionModel,
 )
 from diffusers.utils import is_xformers_available, logging
-from diffusers.utils.testing_utils import require_accelerator, torch_device
 
+from ...testing_utils import require_accelerator, torch_device
 from ..pipeline_params import TEXT_TO_IMAGE_BATCH_PARAMS, TEXT_TO_IMAGE_CALLBACK_CFG_PARAMS, TEXT_TO_IMAGE_PARAMS
 from ..test_pipelines_common import (
     IPAdapterTesterMixin,

tests/pipelines/animatediff/test_animatediff_sparsectrl.py

@@ -20,8 +20,8 @@ from diffusers import (
 )
 from diffusers.utils import logging
 from diffusers.utils.import_utils import is_xformers_available
-from diffusers.utils.testing_utils import require_accelerator, torch_device
 
+from ...testing_utils import require_accelerator, torch_device
 from ..pipeline_params import TEXT_TO_IMAGE_BATCH_PARAMS, TEXT_TO_IMAGE_PARAMS
 from ..test_pipelines_common import (
     IPAdapterTesterMixin,

tests/pipelines/animatediff/test_animatediff_video2video.py

@@ -19,8 +19,8 @@ from diffusers import (
 )
 from diffusers.models.attention import FreeNoiseTransformerBlock
 from diffusers.utils import is_xformers_available, logging
-from diffusers.utils.testing_utils import require_accelerator, torch_device
 
+from ...testing_utils import require_accelerator, torch_device
 from ..pipeline_params import TEXT_TO_IMAGE_PARAMS, VIDEO_TO_VIDEO_BATCH_PARAMS
 from ..test_pipelines_common import IPAdapterTesterMixin, PipelineFromPipeTesterMixin, PipelineTesterMixin
 

tests/pipelines/animatediff/test_animatediff_video2video_controlnet.py

@@ -20,8 +20,8 @@ from diffusers import (
 )
 from diffusers.models.attention import FreeNoiseTransformerBlock
 from diffusers.utils import is_xformers_available, logging
-from diffusers.utils.testing_utils import require_accelerator, torch_device
 
+from ...testing_utils import require_accelerator, torch_device
 from ..pipeline_params import TEXT_TO_IMAGE_PARAMS, VIDEO_TO_VIDEO_BATCH_PARAMS
 from ..test_pipelines_common import IPAdapterTesterMixin, PipelineFromPipeTesterMixin, PipelineTesterMixin
 

tests/pipelines/audioldm2/test_audioldm2.py

@@ -46,14 +46,14 @@ from diffusers import (
     PNDMScheduler,
 )
 from diffusers.utils import is_transformers_version
-from diffusers.utils.testing_utils import (
+
+from ...testing_utils import (
     backend_empty_cache,
     enable_full_determinism,
     is_torch_version,
     nightly,
     torch_device,
 )
-
 from ..pipeline_params import TEXT_TO_AUDIO_BATCH_PARAMS, TEXT_TO_AUDIO_PARAMS
 from ..test_pipelines_common import PipelineTesterMixin
 

tests/pipelines/bria/test_pipeline_bria.py

@@ -27,7 +27,11 @@ from diffusers import (
     FlowMatchEulerDiscreteScheduler,
 )
 from diffusers.pipelines.bria import BriaPipeline
-from diffusers.utils.testing_utils import (
+
+# from ..test_pipelines_common import PipelineTesterMixin, check_qkv_fused_layers_exist
+from tests.pipelines.test_pipelines_common import PipelineTesterMixin, to_np
+
+from ...testing_utils import (
     backend_empty_cache,
     enable_full_determinism,
     numpy_cosine_similarity_distance,
@@ -36,9 +40,6 @@ from diffusers.utils.testing_utils import (
     torch_device,
 )
 
-# from ..test_pipelines_common import PipelineTesterMixin, check_qkv_fused_layers_exist
-from tests.pipelines.test_pipelines_common import PipelineTesterMixin, to_np
-
 
 enable_full_determinism()