From bdd4201fdba05245342e6013431f5209e0bcc773 Mon Sep 17 00:00:00 2001 From: Fanli Lin Date: Fri, 13 Dec 2024 21:33:45 +0800 Subject: [PATCH] [tests] fix "Tester object has no attribute '_testMethodName'" (#34910) * add more cases * fix method not found in unittest Signed-off-by: Lin, Fanli * fix more cases * add more models * add all * no unittest.case * remove for oneformer * fix style --------- Signed-off-by: Lin, Fanli --- .../test_feature_extraction_audio_spectrogram_transformer.py | 2 +- tests/models/beit/test_image_processing_beit.py | 2 +- tests/models/clap/test_feature_extraction_clap.py | 2 +- tests/models/clip/test_image_processing_clip.py | 2 +- tests/models/clvp/test_feature_extraction_clvp.py | 2 +- .../conditional_detr/test_image_processing_conditional_detr.py | 2 +- tests/models/dac/test_feature_extraction_dac.py | 2 +- tests/models/encodec/test_feature_extraction_encodec.py | 2 +- .../grounding_dino/test_image_processing_grounding_dino.py | 2 +- tests/models/idefics/test_image_processing_idefics.py | 2 +- tests/models/idefics2/test_image_processing_idefics2.py | 2 +- tests/models/llava_next/test_image_processing_llava_next.py | 2 +- .../llava_next_video/test_image_processing_llava_next_video.py | 2 +- .../llava_onevision/test_image_processing_llava_onevision.py | 2 +- tests/models/markuplm/test_feature_extraction_markuplm.py | 2 +- tests/models/mask2former/test_image_processing_mask2former.py | 2 +- tests/models/maskformer/test_image_processing_maskformer.py | 2 +- .../musicgen_melody/test_feature_extraction_musicgen_melody.py | 2 +- tests/models/oneformer/test_image_processing_oneformer.py | 2 +- tests/models/oneformer/test_processor_oneformer.py | 2 +- tests/models/pix2struct/test_image_processing_pix2struct.py | 2 +- tests/models/pixtral/test_image_processing_pixtral.py | 2 +- tests/models/pop2piano/test_feature_extraction_pop2piano.py | 2 +- tests/models/qwen2_vl/test_image_processing_qwen2_vl.py | 2 +- .../models/seamless_m4t/test_feature_extraction_seamless_m4t.py | 2 +- tests/models/segformer/test_image_processing_segformer.py | 2 +- tests/models/seggpt/test_image_processing_seggpt.py | 2 +- .../speech_to_text/test_feature_extraction_speech_to_text.py | 2 +- tests/models/speecht5/test_feature_extraction_speecht5.py | 2 +- tests/models/superpoint/test_image_processing_superpoint.py | 2 +- tests/models/univnet/test_feature_extraction_univnet.py | 2 +- tests/models/wav2vec2/test_feature_extraction_wav2vec2.py | 2 +- tests/models/whisper/test_feature_extraction_whisper.py | 2 +- tests/models/yolos/test_image_processing_yolos.py | 2 +- 34 files changed, 34 insertions(+), 34 deletions(-) diff --git a/tests/models/audio_spectrogram_transformer/test_feature_extraction_audio_spectrogram_transformer.py b/tests/models/audio_spectrogram_transformer/test_feature_extraction_audio_spectrogram_transformer.py index fbe2509086..ff33de487d 100644 --- a/tests/models/audio_spectrogram_transformer/test_feature_extraction_audio_spectrogram_transformer.py +++ b/tests/models/audio_spectrogram_transformer/test_feature_extraction_audio_spectrogram_transformer.py @@ -50,7 +50,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None): return values -class ASTFeatureExtractionTester(unittest.TestCase): +class ASTFeatureExtractionTester: def __init__( self, parent, diff --git a/tests/models/beit/test_image_processing_beit.py b/tests/models/beit/test_image_processing_beit.py index 526a78a563..58175c6fe1 100644 --- a/tests/models/beit/test_image_processing_beit.py +++ b/tests/models/beit/test_image_processing_beit.py @@ -33,7 +33,7 @@ if is_vision_available(): from transformers import BeitImageProcessor -class BeitImageProcessingTester(unittest.TestCase): +class BeitImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/clap/test_feature_extraction_clap.py b/tests/models/clap/test_feature_extraction_clap.py index d0e913df82..0d6c00b79d 100644 --- a/tests/models/clap/test_feature_extraction_clap.py +++ b/tests/models/clap/test_feature_extraction_clap.py @@ -53,7 +53,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None): @require_torch @require_torchaudio # Copied from tests.models.whisper.test_feature_extraction_whisper.WhisperFeatureExtractionTester with Whisper->Clap -class ClapFeatureExtractionTester(unittest.TestCase): +class ClapFeatureExtractionTester: def __init__( self, parent, diff --git a/tests/models/clip/test_image_processing_clip.py b/tests/models/clip/test_image_processing_clip.py index 740399d13f..ef4fdc819b 100644 --- a/tests/models/clip/test_image_processing_clip.py +++ b/tests/models/clip/test_image_processing_clip.py @@ -26,7 +26,7 @@ if is_vision_available(): from transformers import CLIPImageProcessor -class CLIPImageProcessingTester(unittest.TestCase): +class CLIPImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/clvp/test_feature_extraction_clvp.py b/tests/models/clvp/test_feature_extraction_clvp.py index 1f059ca469..b57cb65ebb 100644 --- a/tests/models/clvp/test_feature_extraction_clvp.py +++ b/tests/models/clvp/test_feature_extraction_clvp.py @@ -57,7 +57,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None): @require_torch -class ClvpFeatureExtractionTester(unittest.TestCase): +class ClvpFeatureExtractionTester: def __init__( self, parent, diff --git a/tests/models/conditional_detr/test_image_processing_conditional_detr.py b/tests/models/conditional_detr/test_image_processing_conditional_detr.py index 32b135bcd2..4e46161a7b 100644 --- a/tests/models/conditional_detr/test_image_processing_conditional_detr.py +++ b/tests/models/conditional_detr/test_image_processing_conditional_detr.py @@ -35,7 +35,7 @@ if is_vision_available(): from transformers import ConditionalDetrImageProcessor -class ConditionalDetrImageProcessingTester(unittest.TestCase): +class ConditionalDetrImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/dac/test_feature_extraction_dac.py b/tests/models/dac/test_feature_extraction_dac.py index 019a4f07c6..598a7c725e 100644 --- a/tests/models/dac/test_feature_extraction_dac.py +++ b/tests/models/dac/test_feature_extraction_dac.py @@ -51,7 +51,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None): @require_torch # Copied from transformers.tests.encodec.test_feature_extraction_dac.EncodecFeatureExtractionTester with Encodec->Dac -class DacFeatureExtractionTester(unittest.TestCase): +class DacFeatureExtractionTester: # Ignore copy def __init__( self, diff --git a/tests/models/encodec/test_feature_extraction_encodec.py b/tests/models/encodec/test_feature_extraction_encodec.py index e56517ac41..112f1022c0 100644 --- a/tests/models/encodec/test_feature_extraction_encodec.py +++ b/tests/models/encodec/test_feature_extraction_encodec.py @@ -50,7 +50,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None): @require_torch -class EnCodecFeatureExtractionTester(unittest.TestCase): +class EnCodecFeatureExtractionTester: def __init__( self, parent, diff --git a/tests/models/grounding_dino/test_image_processing_grounding_dino.py b/tests/models/grounding_dino/test_image_processing_grounding_dino.py index bb8b9272ef..5cc1e6c232 100644 --- a/tests/models/grounding_dino/test_image_processing_grounding_dino.py +++ b/tests/models/grounding_dino/test_image_processing_grounding_dino.py @@ -37,7 +37,7 @@ if is_vision_available(): from transformers import GroundingDinoImageProcessor -class GroundingDinoImageProcessingTester(unittest.TestCase): +class GroundingDinoImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/idefics/test_image_processing_idefics.py b/tests/models/idefics/test_image_processing_idefics.py index 2f7a8993df..ad20888157 100644 --- a/tests/models/idefics/test_image_processing_idefics.py +++ b/tests/models/idefics/test_image_processing_idefics.py @@ -36,7 +36,7 @@ if is_vision_available(): from transformers import IdeficsImageProcessor -class IdeficsImageProcessingTester(unittest.TestCase): +class IdeficsImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/idefics2/test_image_processing_idefics2.py b/tests/models/idefics2/test_image_processing_idefics2.py index 624fdd6c98..bf9634b398 100644 --- a/tests/models/idefics2/test_image_processing_idefics2.py +++ b/tests/models/idefics2/test_image_processing_idefics2.py @@ -34,7 +34,7 @@ if is_torch_available(): import torch -class Idefics2ImageProcessingTester(unittest.TestCase): +class Idefics2ImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/llava_next/test_image_processing_llava_next.py b/tests/models/llava_next/test_image_processing_llava_next.py index fc399298c3..4b3f5e0dd3 100644 --- a/tests/models/llava_next/test_image_processing_llava_next.py +++ b/tests/models/llava_next/test_image_processing_llava_next.py @@ -34,7 +34,7 @@ if is_vision_available(): from transformers import LlavaNextImageProcessor -class LlavaNextImageProcessingTester(unittest.TestCase): +class LlavaNextImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/llava_next_video/test_image_processing_llava_next_video.py b/tests/models/llava_next_video/test_image_processing_llava_next_video.py index 8c525fa256..385475c262 100644 --- a/tests/models/llava_next_video/test_image_processing_llava_next_video.py +++ b/tests/models/llava_next_video/test_image_processing_llava_next_video.py @@ -33,7 +33,7 @@ if is_vision_available(): from transformers import LlavaNextVideoImageProcessor -class LlavaNextVideoProcessingTester(unittest.TestCase): +class LlavaNextVideoProcessingTester: def __init__( self, parent, diff --git a/tests/models/llava_onevision/test_image_processing_llava_onevision.py b/tests/models/llava_onevision/test_image_processing_llava_onevision.py index 47b6ef86c5..f392f2b895 100644 --- a/tests/models/llava_onevision/test_image_processing_llava_onevision.py +++ b/tests/models/llava_onevision/test_image_processing_llava_onevision.py @@ -33,7 +33,7 @@ if is_vision_available(): from transformers import LlavaOnevisionImageProcessor, LlavaOnevisionVideoProcessor -class LlavaOnevisionImageProcessingTester(unittest.TestCase): +class LlavaOnevisionImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/markuplm/test_feature_extraction_markuplm.py b/tests/models/markuplm/test_feature_extraction_markuplm.py index 4541cb9480..381483d655 100644 --- a/tests/models/markuplm/test_feature_extraction_markuplm.py +++ b/tests/models/markuplm/test_feature_extraction_markuplm.py @@ -26,7 +26,7 @@ if is_bs4_available(): from transformers import MarkupLMFeatureExtractor -class MarkupLMFeatureExtractionTester(unittest.TestCase): +class MarkupLMFeatureExtractionTester: def __init__(self, parent): self.parent = parent diff --git a/tests/models/mask2former/test_image_processing_mask2former.py b/tests/models/mask2former/test_image_processing_mask2former.py index 7468c3fd47..b298336a81 100644 --- a/tests/models/mask2former/test_image_processing_mask2former.py +++ b/tests/models/mask2former/test_image_processing_mask2former.py @@ -39,7 +39,7 @@ if is_vision_available(): from PIL import Image -class Mask2FormerImageProcessingTester(unittest.TestCase): +class Mask2FormerImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/maskformer/test_image_processing_maskformer.py b/tests/models/maskformer/test_image_processing_maskformer.py index 23e517a326..8b3c7db762 100644 --- a/tests/models/maskformer/test_image_processing_maskformer.py +++ b/tests/models/maskformer/test_image_processing_maskformer.py @@ -38,7 +38,7 @@ if is_vision_available(): from PIL import Image -class MaskFormerImageProcessingTester(unittest.TestCase): +class MaskFormerImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/musicgen_melody/test_feature_extraction_musicgen_melody.py b/tests/models/musicgen_melody/test_feature_extraction_musicgen_melody.py index 697e3fb146..bdd1cb1e12 100644 --- a/tests/models/musicgen_melody/test_feature_extraction_musicgen_melody.py +++ b/tests/models/musicgen_melody/test_feature_extraction_musicgen_melody.py @@ -69,7 +69,7 @@ def get_bip_bip(bip_duration=0.125, duration=0.5, sample_rate=32000): @require_torch @require_torchaudio -class MusicgenMelodyFeatureExtractionTester(unittest.TestCase): +class MusicgenMelodyFeatureExtractionTester: def __init__( self, parent, diff --git a/tests/models/oneformer/test_image_processing_oneformer.py b/tests/models/oneformer/test_image_processing_oneformer.py index 853bf241dd..7ac52f76b4 100644 --- a/tests/models/oneformer/test_image_processing_oneformer.py +++ b/tests/models/oneformer/test_image_processing_oneformer.py @@ -39,7 +39,7 @@ if is_vision_available(): from PIL import Image -class OneFormerImageProcessorTester(unittest.TestCase): +class OneFormerImageProcessorTester: def __init__( self, parent, diff --git a/tests/models/oneformer/test_processor_oneformer.py b/tests/models/oneformer/test_processor_oneformer.py index 3a8a378b49..dae50040ec 100644 --- a/tests/models/oneformer/test_processor_oneformer.py +++ b/tests/models/oneformer/test_processor_oneformer.py @@ -59,7 +59,7 @@ def prepare_metadata(class_info_file, repo_path="shi-labs/oneformer_demo"): return metadata -class OneFormerProcessorTester(unittest.TestCase): +class OneFormerProcessorTester: def __init__( self, parent, diff --git a/tests/models/pix2struct/test_image_processing_pix2struct.py b/tests/models/pix2struct/test_image_processing_pix2struct.py index 2d5616b5b7..6b12b3827d 100644 --- a/tests/models/pix2struct/test_image_processing_pix2struct.py +++ b/tests/models/pix2struct/test_image_processing_pix2struct.py @@ -34,7 +34,7 @@ if is_vision_available(): from transformers import Pix2StructImageProcessor -class Pix2StructImageProcessingTester(unittest.TestCase): +class Pix2StructImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/pixtral/test_image_processing_pixtral.py b/tests/models/pixtral/test_image_processing_pixtral.py index 8b49b5aa60..a45ead5061 100644 --- a/tests/models/pixtral/test_image_processing_pixtral.py +++ b/tests/models/pixtral/test_image_processing_pixtral.py @@ -38,7 +38,7 @@ if is_vision_available(): from transformers import PixtralImageProcessorFast -class PixtralImageProcessingTester(unittest.TestCase): +class PixtralImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/pop2piano/test_feature_extraction_pop2piano.py b/tests/models/pop2piano/test_feature_extraction_pop2piano.py index c676614797..6b4b1b987a 100644 --- a/tests/models/pop2piano/test_feature_extraction_pop2piano.py +++ b/tests/models/pop2piano/test_feature_extraction_pop2piano.py @@ -48,7 +48,7 @@ if requirements_available: from transformers import Pop2PianoFeatureExtractor -class Pop2PianoFeatureExtractionTester(unittest.TestCase): +class Pop2PianoFeatureExtractionTester: def __init__( self, parent, diff --git a/tests/models/qwen2_vl/test_image_processing_qwen2_vl.py b/tests/models/qwen2_vl/test_image_processing_qwen2_vl.py index d69addb9a1..a6004349b4 100644 --- a/tests/models/qwen2_vl/test_image_processing_qwen2_vl.py +++ b/tests/models/qwen2_vl/test_image_processing_qwen2_vl.py @@ -34,7 +34,7 @@ if is_vision_available(): from transformers import Qwen2VLImageProcessor -class Qwen2VLImageProcessingTester(unittest.TestCase): +class Qwen2VLImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/seamless_m4t/test_feature_extraction_seamless_m4t.py b/tests/models/seamless_m4t/test_feature_extraction_seamless_m4t.py index 8830660c09..7c13f97b64 100644 --- a/tests/models/seamless_m4t/test_feature_extraction_seamless_m4t.py +++ b/tests/models/seamless_m4t/test_feature_extraction_seamless_m4t.py @@ -52,7 +52,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None): @require_torch -class SeamlessM4TFeatureExtractionTester(unittest.TestCase): +class SeamlessM4TFeatureExtractionTester: def __init__( self, parent, diff --git a/tests/models/segformer/test_image_processing_segformer.py b/tests/models/segformer/test_image_processing_segformer.py index 2239930001..dba2de7e48 100644 --- a/tests/models/segformer/test_image_processing_segformer.py +++ b/tests/models/segformer/test_image_processing_segformer.py @@ -33,7 +33,7 @@ if is_vision_available(): from transformers import SegformerImageProcessor -class SegformerImageProcessingTester(unittest.TestCase): +class SegformerImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/seggpt/test_image_processing_seggpt.py b/tests/models/seggpt/test_image_processing_seggpt.py index f79b7ea443..74e78f0082 100644 --- a/tests/models/seggpt/test_image_processing_seggpt.py +++ b/tests/models/seggpt/test_image_processing_seggpt.py @@ -35,7 +35,7 @@ if is_vision_available(): from transformers import SegGptImageProcessor -class SegGptImageProcessingTester(unittest.TestCase): +class SegGptImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/speech_to_text/test_feature_extraction_speech_to_text.py b/tests/models/speech_to_text/test_feature_extraction_speech_to_text.py index 9023e8467f..2a4ad08949 100644 --- a/tests/models/speech_to_text/test_feature_extraction_speech_to_text.py +++ b/tests/models/speech_to_text/test_feature_extraction_speech_to_text.py @@ -48,7 +48,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None): @require_torch @require_torchaudio -class Speech2TextFeatureExtractionTester(unittest.TestCase): +class Speech2TextFeatureExtractionTester: def __init__( self, parent, diff --git a/tests/models/speecht5/test_feature_extraction_speecht5.py b/tests/models/speecht5/test_feature_extraction_speecht5.py index 5ec632e7e7..70d60f9223 100644 --- a/tests/models/speecht5/test_feature_extraction_speecht5.py +++ b/tests/models/speecht5/test_feature_extraction_speecht5.py @@ -50,7 +50,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None): @require_torch -class SpeechT5FeatureExtractionTester(unittest.TestCase): +class SpeechT5FeatureExtractionTester: def __init__( self, parent, diff --git a/tests/models/superpoint/test_image_processing_superpoint.py b/tests/models/superpoint/test_image_processing_superpoint.py index c2eae87200..e11fd08422 100644 --- a/tests/models/superpoint/test_image_processing_superpoint.py +++ b/tests/models/superpoint/test_image_processing_superpoint.py @@ -33,7 +33,7 @@ if is_vision_available(): from transformers import SuperPointImageProcessor -class SuperPointImageProcessingTester(unittest.TestCase): +class SuperPointImageProcessingTester: def __init__( self, parent, diff --git a/tests/models/univnet/test_feature_extraction_univnet.py b/tests/models/univnet/test_feature_extraction_univnet.py index dfa335d153..2917d206df 100644 --- a/tests/models/univnet/test_feature_extraction_univnet.py +++ b/tests/models/univnet/test_feature_extraction_univnet.py @@ -50,7 +50,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None): return values -class UnivNetFeatureExtractionTester(unittest.TestCase): +class UnivNetFeatureExtractionTester: def __init__( self, parent, diff --git a/tests/models/wav2vec2/test_feature_extraction_wav2vec2.py b/tests/models/wav2vec2/test_feature_extraction_wav2vec2.py index 29e4bf3e28..2a92ce3ac3 100644 --- a/tests/models/wav2vec2/test_feature_extraction_wav2vec2.py +++ b/tests/models/wav2vec2/test_feature_extraction_wav2vec2.py @@ -44,7 +44,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None): return values -class Wav2Vec2FeatureExtractionTester(unittest.TestCase): +class Wav2Vec2FeatureExtractionTester: def __init__( self, parent, diff --git a/tests/models/whisper/test_feature_extraction_whisper.py b/tests/models/whisper/test_feature_extraction_whisper.py index a8295542f4..4b2353bce0 100644 --- a/tests/models/whisper/test_feature_extraction_whisper.py +++ b/tests/models/whisper/test_feature_extraction_whisper.py @@ -50,7 +50,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None): return values -class WhisperFeatureExtractionTester(unittest.TestCase): +class WhisperFeatureExtractionTester: def __init__( self, parent, diff --git a/tests/models/yolos/test_image_processing_yolos.py b/tests/models/yolos/test_image_processing_yolos.py index 67508532e9..55a4be5c09 100644 --- a/tests/models/yolos/test_image_processing_yolos.py +++ b/tests/models/yolos/test_image_processing_yolos.py @@ -36,7 +36,7 @@ if is_vision_available(): from transformers import YolosImageProcessor -class YolosImageProcessingTester(unittest.TestCase): +class YolosImageProcessingTester: def __init__( self, parent,