deblan-mirror/buzz
1
0
Fork 0
mirror of https://github.com/chidiwilliams/buzz.git synced 2026-03-14 14:45:46 +01:00
Code Issues Projects Releases Packages Wiki Activity

Add black formatting (#571)

Browse source
This commit is contained in:
Chidi Williams 2023-08-18 23:32:18 +01:00 • committed by GitHub
commit c498e60949
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
66 changed files with 2556 additions and 1384 deletions
Download patch file Download diff file Expand all files Collapse all files

31
tests/cache_test.py
View file

@ -1,16 +1,31 @@
from buzz.cache import TasksCache
from buzz.transcriber import (FileTranscriptionOptions, FileTranscriptionTask,
TranscriptionOptions)
from buzz.transcriber import (
FileTranscriptionOptions,
FileTranscriptionTask,
TranscriptionOptions,
)
class TestTasksCache:
def test_should_save_and_load(self, tmp_path):
cache = TasksCache(cache_dir=str(tmp_path))
tasks = [FileTranscriptionTask(file_path='1.mp3', transcription_options=TranscriptionOptions(),
file_transcription_options=FileTranscriptionOptions(file_paths=['1.mp3']),
model_path=''),
FileTranscriptionTask(file_path='2.mp3', transcription_options=TranscriptionOptions(),
file_transcription_options=FileTranscriptionOptions(file_paths=['2.mp3']),
model_path='')]
tasks = [
FileTranscriptionTask(
file_path="1.mp3",
transcription_options=TranscriptionOptions(),
file_transcription_options=FileTranscriptionOptions(
file_paths=["1.mp3"]
),
model_path="",
),
FileTranscriptionTask(
file_path="2.mp3",
transcription_options=TranscriptionOptions(),
file_transcription_options=FileTranscriptionOptions(
file_paths=["2.mp3"]
),
model_path="",
),
]
cache.save(tasks)
assert cache.load() == tasks