diff --git a/test/torchcodec/decoders.py b/test/torchcodec/decoders.py new file mode 100644 index 0000000000..94f2d8c8c1 --- /dev/null +++ b/test/torchcodec/decoders.py @@ -0,0 +1,17 @@ +import test.torchaudio_unittest.common_utils.wav_utils as wav_utils + +class AudioDecoder: + def __init__(self, uri): + self.uri = uri + + def get_all_samples(self): + return wav_utils.load_wav(self.uri) + + +class AudioEncoder: + def __init__(self, data, sample_rate): + self.data = data + self.sample_rate = sample_rate + + def to_file(self, uri, bit_rate=None): + return wav_utils.save_wav(uri, self.data, self.sample_rate) diff --git a/test/torchcodec/encoders.py b/test/torchcodec/encoders.py new file mode 100644 index 0000000000..5e9cc54968 --- /dev/null +++ b/test/torchcodec/encoders.py @@ -0,0 +1,10 @@ +import torchaudio_unittest.common_utils.wav_utils as wav_utils + +class AudioEncoder: + def __init__(self, data, sample_rate): + print("BEING CALLED") + self.data = data + self.sample_rate = sample_rate + + def to_file(self, uri, bit_rate=None): + return wav_utils.save_wav(uri, self.data, self.sample_rate)