[Test refactor 1/5] Per-folder tests reorganization (#15725)

* Per-folder tests reorganization
Co-authored-by: sgugger <sylvain.gugger@gmail.com>
Co-authored-by: Stas Bekman <stas@stason.org>

tests/test_modeling_trocr.py → tests/trocr/test_modeling_trocr.py

@@ -19,9 +19,9 @@ import unittest
 from transformers import TrOCRConfig
 from transformers.testing_utils import is_torch_available, require_torch, torch_device
 
-from .test_configuration_common import ConfigTester
-from .test_generation_utils import GenerationTesterMixin
-from .test_modeling_common import ModelTesterMixin, ids_tensor
+from ..generation.test_generation_utils import GenerationTesterMixin
+from ..test_configuration_common import ConfigTester
+from ..test_modeling_common import ModelTesterMixin, ids_tensor
 
 
 if is_torch_available():

tests/test_modeling_unispeech.py → tests/unispeech/test_modeling_unispeech.py

@@ -21,12 +21,17 @@ import numpy as np
 import pytest
 from datasets import load_dataset
 
-from tests.test_modeling_common import floats_tensor, ids_tensor, random_attention_mask
 from transformers import UniSpeechConfig, is_torch_available
 from transformers.testing_utils import require_soundfile, require_torch, slow, torch_device
 
-from .test_configuration_common import ConfigTester
-from .test_modeling_common import ModelTesterMixin, _config_zero_init
+from ..test_configuration_common import ConfigTester
+from ..test_modeling_common import (
+    ModelTesterMixin,
+    _config_zero_init,
+    floats_tensor,
+    ids_tensor,
+    random_attention_mask,
+)
 
 
 if is_torch_available():

tests/test_modeling_unispeech_sat.py → tests/unispeech_sat/test_modeling_unispeech_sat.py

@@ -21,12 +21,17 @@ import numpy as np
 import pytest
 from datasets import load_dataset
 
-from tests.test_modeling_common import floats_tensor, ids_tensor, random_attention_mask
 from transformers import UniSpeechSatConfig, is_torch_available
 from transformers.testing_utils import require_soundfile, require_torch, slow, torch_device
 
-from .test_configuration_common import ConfigTester
-from .test_modeling_common import ModelTesterMixin, _config_zero_init
+from ..test_configuration_common import ConfigTester
+from ..test_modeling_common import (
+    ModelTesterMixin,
+    _config_zero_init,
+    floats_tensor,
+    ids_tensor,
+    random_attention_mask,
+)
 
 
 if is_torch_available():

tests/test_add_new_model_like.py → tests/utils/test_add_new_model_like.py

@@ -1008,7 +1008,7 @@ else:
 """
 
         with tempfile.TemporaryDirectory() as tmp_dir:
-            file_name = os.path.join(tmp_dir, "__init__.py")
+            file_name = os.path.join(tmp_dir, "../__init__.py")
 
             self.init_file(file_name, test_init)
             clean_frameworks_in_init(file_name, keep_processing=False)
@@ -1156,7 +1156,7 @@ else:
 """
 
         with tempfile.TemporaryDirectory() as tmp_dir:
-            file_name = os.path.join(tmp_dir, "__init__.py")
+            file_name = os.path.join(tmp_dir, "../__init__.py")
 
             self.init_file(file_name, test_init)
             clean_frameworks_in_init(file_name, keep_processing=False)

tests/test_modeling_tf_core.py → tests/utils/test_modeling_tf_core.py

@@ -23,7 +23,7 @@ from transformers import is_tf_available
 from transformers.models.auto import get_values
 from transformers.testing_utils import _tf_gpu_memory_limit, require_tf, slow
 
-from .test_modeling_tf_common import ids_tensor
+from ..test_modeling_tf_common import ids_tensor
 
 
 if is_tf_available():