|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import os |
|
import unittest |
|
|
|
from transformers import AutoFeatureExtractor, Wav2Vec2FeatureExtractor |
|
|
|
|
|
SAMPLE_FEATURE_EXTRACTION_CONFIG_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "fixtures") |
|
SAMPLE_FEATURE_EXTRACTION_CONFIG = os.path.join( |
|
os.path.dirname(os.path.abspath(__file__)), "fixtures/dummy_feature_extractor_config.json" |
|
) |
|
|
|
|
|
class AutoFeatureExtractorTest(unittest.TestCase): |
|
def test_feature_extractor_from_model_shortcut(self): |
|
config = AutoFeatureExtractor.from_pretrained("facebook/wav2vec2-base-960h") |
|
self.assertIsInstance(config, Wav2Vec2FeatureExtractor) |
|
|
|
def test_feature_extractor_from_local_directory(self): |
|
config = AutoFeatureExtractor.from_pretrained(SAMPLE_FEATURE_EXTRACTION_CONFIG_DIR) |
|
self.assertIsInstance(config, Wav2Vec2FeatureExtractor) |
|
|
|
def test_feature_extractor_from_local_file(self): |
|
config = AutoFeatureExtractor.from_pretrained(SAMPLE_FEATURE_EXTRACTION_CONFIG) |
|
self.assertIsInstance(config, Wav2Vec2FeatureExtractor) |
|
|