From 2e25279e5382f591272f2b5d01286cc11d976f79 Mon Sep 17 00:00:00 2001 From: Sam Anklesaria Date: Fri, 18 Jul 2025 19:57:20 +0000 Subject: [PATCH] Add torchcodec mock with wav loading and saving --- test/torchcodec/decoders.py | 17 +++++++++++++++++ test/torchcodec/encoders.py | 10 ++++++++++ 2 files changed, 27 insertions(+) create mode 100644 test/torchcodec/decoders.py create mode 100644 test/torchcodec/encoders.py 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)