def test_get_audio_set():
from utilities.utilities import Utilities
u = Utilities()
expected = {'a.wav', 'b.wav', 'c.wav'}
data = ['a.wav', 'foo.txt', 'c.txt', 'b.wav', 'c.wav', 'a.txt', 'b.txt']
result = u.get_audio_set(data)
assert result == expected
dia = False
punct = False
c = None
if only_transcribe:
dia = input('Add Diarization Y/N ')
if dia.lower() == 'y':
c = input('How many speakers (int) ')
configuration.set_diarizationSpeakerCount(int(c))
configuration.set_enableSpeakerDiarization(bool(dia))
else:
print('No diarization')
punct = input('Add Punctuation Y/N? ')
if punct.lower() == 'y':
configuration.set_enableAutomaticPunctuation(True)
audio_set = utilities.get_audio_set(final_file_list)
logger.debug(f'AUDIO SET: {audio_set}')
audio_list = list()
if limit:
count = 0
while count < limit:
audio_list.append(audio_set.pop())
count+=1
string = f'Limit to {limit} audio files'
print(string)
logger.debug(string)
else:
audio_list = list(audio_set)
# Prompt for confirmation
for item in audio_list: