[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_flax_roberta.py → tests/roberta/test_modeling_flax_roberta.py

@@ -19,7 +19,7 @@ import numpy as np
 from transformers import RobertaConfig, is_flax_available
 from transformers.testing_utils import require_flax, slow
 
-from .test_modeling_flax_common import FlaxModelTesterMixin, ids_tensor, random_attention_mask
+from ..test_modeling_flax_common import FlaxModelTesterMixin, ids_tensor, random_attention_mask
 
 
 if is_flax_available():

tests/test_modeling_roberta.py → tests/roberta/test_modeling_roberta.py

@@ -20,9 +20,9 @@ from copy import deepcopy
 from transformers import RobertaConfig, is_torch_available
 from transformers.testing_utils import TestCasePlus, require_torch, slow, torch_device
 
-from .test_configuration_common import ConfigTester
-from .test_generation_utils import GenerationTesterMixin
-from .test_modeling_common import ModelTesterMixin, floats_tensor, ids_tensor, random_attention_mask
+from ..generation.test_generation_utils import GenerationTesterMixin
+from ..test_configuration_common import ConfigTester
+from ..test_modeling_common import ModelTesterMixin, floats_tensor, ids_tensor, random_attention_mask
 
 
 if is_torch_available():

tests/test_modeling_tf_roberta.py → tests/roberta/test_modeling_tf_roberta.py

@@ -19,8 +19,8 @@ import unittest
 from transformers import RobertaConfig, is_tf_available
 from transformers.testing_utils import require_sentencepiece, require_tf, require_tokenizers, slow
 
-from .test_configuration_common import ConfigTester
-from .test_modeling_tf_common import TFModelTesterMixin, floats_tensor, ids_tensor
+from ..test_configuration_common import ConfigTester
+from ..test_modeling_tf_common import TFModelTesterMixin, floats_tensor, ids_tensor
 
 
 if is_tf_available():

tests/test_tokenization_roberta.py → tests/roberta/test_tokenization_roberta.py

@@ -23,7 +23,7 @@ from transformers import AddedToken, RobertaTokenizer, RobertaTokenizerFast
 from transformers.models.roberta.tokenization_roberta import VOCAB_FILES_NAMES
 from transformers.testing_utils import require_tokenizers, slow
 
-from .test_tokenization_common import TokenizerTesterMixin
+from ..test_tokenization_common import TokenizerTesterMixin
 
 
 @require_tokenizers

tests/test_modeling_flax_roformer.py → tests/roformer/test_modeling_flax_roformer.py

@@ -19,7 +19,7 @@ import numpy as np
 from transformers import RoFormerConfig, is_flax_available
 from transformers.testing_utils import require_flax, slow
 
-from .test_modeling_flax_common import FlaxModelTesterMixin, ids_tensor, random_attention_mask
+from ..test_modeling_flax_common import FlaxModelTesterMixin, ids_tensor, random_attention_mask
 
 
 if is_flax_available():

tests/test_modeling_roformer.py → tests/roformer/test_modeling_roformer.py

@@ -17,12 +17,11 @@
 
 import unittest
 
-from tests.test_modeling_common import floats_tensor
 from transformers import RoFormerConfig, is_torch_available
 from transformers.testing_utils import require_torch, slow, torch_device
 
-from .test_configuration_common import ConfigTester
-from .test_modeling_common import ModelTesterMixin, ids_tensor, random_attention_mask
+from ..test_configuration_common import ConfigTester
+from ..test_modeling_common import ModelTesterMixin, floats_tensor, ids_tensor, random_attention_mask
 
 
 if is_torch_available():

tests/test_modeling_tf_roformer.py → tests/roformer/test_modeling_tf_roformer.py

@@ -19,8 +19,8 @@ import unittest
 from transformers import RoFormerConfig, is_tf_available
 from transformers.testing_utils import require_tf, slow
 
-from .test_configuration_common import ConfigTester
-from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
+from ..test_configuration_common import ConfigTester
+from ..test_modeling_tf_common import TFModelTesterMixin, ids_tensor
 
 
 if is_tf_available():

tests/test_tokenization_roformer.py → tests/roformer/test_tokenization_roformer.py

@@ -18,7 +18,7 @@ import unittest
 from transformers import RoFormerTokenizer, RoFormerTokenizerFast
 from transformers.testing_utils import require_rjieba, require_tokenizers
 
-from .test_tokenization_common import TokenizerTesterMixin
+from ..test_tokenization_common import TokenizerTesterMixin
 
 
 @require_rjieba

tests/test_feature_extraction_segformer.py → tests/segformer/test_feature_extraction_segformer.py

@@ -22,7 +22,7 @@ from datasets import load_dataset
 from transformers.file_utils import is_torch_available, is_vision_available
 from transformers.testing_utils import require_torch, require_vision
 
-from .test_feature_extraction_common import FeatureExtractionSavingTestMixin, prepare_image_inputs
+from ..test_feature_extraction_common import FeatureExtractionSavingTestMixin, prepare_image_inputs
 
 
 if is_torch_available():

tests/test_modeling_segformer.py → tests/segformer/test_modeling_segformer.py

@@ -22,8 +22,8 @@ from transformers import is_torch_available, is_vision_available
 from transformers.models.auto import get_values
 from transformers.testing_utils import require_torch, slow, torch_device
 
-from .test_configuration_common import ConfigTester
-from .test_modeling_common import ModelTesterMixin, floats_tensor, ids_tensor
+from ..test_configuration_common import ConfigTester
+from ..test_modeling_common import ModelTesterMixin, floats_tensor, ids_tensor
 
 
 if is_torch_available():

tests/test_modeling_sew.py → tests/sew/test_modeling_sew.py

@@ -20,12 +20,17 @@ import unittest
 
 import pytest
 
-from tests.test_modeling_common import floats_tensor, ids_tensor, random_attention_mask
 from transformers import SEWConfig, 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_sew_d.py → tests/sew_d/test_modeling_sew_d.py

@@ -20,12 +20,17 @@ import unittest
 
 import pytest
 
-from tests.test_modeling_common import floats_tensor, ids_tensor, random_attention_mask
 from transformers import SEWDConfig, 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_speech_encoder_decoder.py → tests/speech_encoder_decoder/test_modeling_speech_encoder_decoder.py

@@ -20,11 +20,11 @@ import unittest
 from transformers import is_torch_available
 from transformers.testing_utils import require_torch, slow, torch_device
 
-from .test_modeling_bert import BertModelTester
-from .test_modeling_common import floats_tensor, ids_tensor, random_attention_mask
-from .test_modeling_speech_to_text import Speech2TextModelTester
-from .test_modeling_speech_to_text_2 import Speech2Text2StandaloneDecoderModelTester
-from .test_modeling_wav2vec2 import Wav2Vec2ModelTester
+from ..bert.test_modeling_bert import BertModelTester
+from ..speech_to_text.test_modeling_speech_to_text import Speech2TextModelTester
+from ..speech_to_text_2.test_modeling_speech_to_text_2 import Speech2Text2StandaloneDecoderModelTester
+from ..test_modeling_common import floats_tensor, ids_tensor, random_attention_mask
+from ..wav2vec2.test_modeling_wav2vec2 import Wav2Vec2ModelTester
 
 
 if is_torch_available():