dnrops
/
transformers.js
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update mel filters unit test

This commit is contained in:
Joshua Lochner 2023-12-03 20:18:09 +02:00
parent bb0d55bdcb
commit 14959112e1
1 changed files with 11 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tests/utils.test.js
View File

@ -19,27 +19,24 @@ describe('Utilities', () => {
// Calculated MEL filters
const calculated_mel_filters = mel_filter_bank(
1 + Math.floor(config.n_fft / 2), // num_frequency_bins
Math.floor(1 + config.n_fft / 2), // num_frequency_bins
config.feature_size, // num_mel_filters
0.0, // min_frequency,
22050.0, // max_frequency
0.0, // min_frequency
8000.0, // max_frequency
config.sampling_rate, // sampling_rate
"slaney", // norm
"slaney", // mel_scale
);
let offset = 0;
let maxdiff = 0;
for (let i = 0; i < original_mel_filters.length; ++i) {
for (let j = 0; j < original_mel_filters[i].length; ++j) {
const expected = original_mel_filters[i][j];
const calculated = calculated_mel_filters.data[offset++];
const diff = Math.abs(expected - calculated);
maxdiff = Math.max(maxdiff, diff);
}
}
const original = original_mel_filters.flat();
const calculated = calculated_mel_filters.flat();
// Compute max difference
const maxdiff = original.reduce((maxdiff, _, i) => {
const diff = Math.abs(original[i] - calculated[i]);
return Math.max(maxdiff, diff);
}, -Infinity);
expect(maxdiff).toBeGreaterThanOrEqual(0);
expect(maxdiff).toBeLessThan(1e-6);
}, MAX_TEST_EXECUTION_TIME);