Skip to content

Commit

Permalink
test wite flac as well
Browse files Browse the repository at this point in the history
  • Loading branch information
MartinCupak committed Jul 31, 2024
1 parent 25c4947 commit e290a66
Showing 1 changed file with 11 additions and 8 deletions.
19 changes: 11 additions & 8 deletions tests/test_calib_meta_dat2wav.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,9 +35,9 @@ def read_calib_parameters(file_path):
return parameters


def calib_dat2wav(rawFileName: str,
calibFileName: str,
calibParamsFileName: str) -> bool:
def calib_dat2wavflac(rawFileName: str,
calibFileName: str,
calibParamsFileName: str) -> bool:

if not os.path.exists(rawFileName):
log.error(f'Raw dat file {rawFileName} not found!')
Expand Down Expand Up @@ -126,6 +126,9 @@ def calib_dat2wav(rawFileName: str,
audiofile.writeMono16bit(wavFileName, sampleRate,
scaledSignal,
essentialMetadata)
wavFileName = audiofile.deriveOutputFileName(rawFileName, 'flac')
audiofile.writeMono16bit(wavFileName, sampleRate,
scaledSignal, essentialMetadata, 'FLAC')
except:
raise AssertionError(f"FAILED: write wave file {wavFileName}")
else:
Expand Down Expand Up @@ -186,7 +189,7 @@ def compare_wav_files(wav1: str, ref1: str):
return are_close


def test_calib_dat2wav():
def test_calib_dat2wavflac():
dat1 = 'tests/data/Rottnest_3154/502DB01D.DAT'
cal1 = 'tests/data/Rottnest_3154/Calib_file/501E9BF5.DAT'
par1 = 'tests/data/Rottnest_3154/Calib_file/Calib_data.TXT'
Expand All @@ -199,9 +202,9 @@ def test_calib_dat2wav():
cal3 = 'tests/data/Portland_3092/Calib_file/4FEACA92.DAT'
par3 = 'tests/data/Portland_3092/Calib_file/Calib_data.TXT'

calib_dat2wav(dat1, cal1, par1)
calib_dat2wav(dat2, cal2, par2)
calib_dat2wav(dat3, cal3, par3)
calib_dat2wavflac(dat1, cal1, par1)
calib_dat2wavflac(dat2, cal2, par2)
calib_dat2wavflac(dat3, cal3, par3)


def nest_calib_dat2wav_reference():
Expand All @@ -220,4 +223,4 @@ def nest_calib_dat2wav_reference():
# seconds resolution is good enough for logging timestamp
datefmt='%Y-%m-%d %H:%M:%S')

test_calib_dat2wav()
test_calib_dat2wavflac()

0 comments on commit e290a66

Please sign in to comment.