Spaces:
Running
Running
| ; | |
| const test = require('node:test'); | |
| const assert = require('node:assert/strict'); | |
| const fs = require('node:fs'); | |
| const zlib = require('node:zlib'); | |
| const path = require('node:path'); | |
| const sound = require('./sound.js'); | |
| const traces = Object.fromEntries(['flint','qwen'].map(model => [model, JSON.parse(zlib.gunzipSync(fs.readFileSync(path.join(__dirname, '../traces/' + model + '-sky-q8.json.gz'))))])); | |
| test('Flint score preserves all selected expert IDs and exact measured weights', () => { | |
| const score = sound.buildScore(traces.flint, 12, .2); | |
| assert.equal(score.events.length, 59); | |
| for (const event of score.events) { | |
| const block = traces.flint.frames[event.index].layers[12]; | |
| assert.deepEqual(event.voices.map(voice => voice.id), block.experts); | |
| assert.deepEqual(event.voices.map(voice => voice.value), block.weights); | |
| event.voices.forEach(voice => assert.equal(voice.amplitude, .11 * Math.sqrt(voice.value))); | |
| } | |
| }); | |
| test('Qwen score uses the exact six largest recurrent head RMS values and a capture-wide scale', () => { | |
| const score = sound.buildScore(traces.qwen, 0, .2); | |
| const maximum = Math.max(...traces.qwen.frames.flatMap(frame => frame.layers[0].heads)); | |
| assert.equal(score.maximum, maximum); | |
| for (const event of score.events) { | |
| const heads = traces.qwen.frames[event.index].layers[0].heads; | |
| const expected = heads.map((value,id) => ({value,id})).sort((a,b) => b.value-a.value || a.id-b.id).slice(0,6); | |
| assert.deepEqual(event.voices.map(({value,id}) => ({value,id})), expected); | |
| event.voices.forEach(voice => { assert.equal(voice.normalized, voice.value / maximum); assert.equal(voice.waveform, 'triangle'); }); | |
| } | |
| }); | |
| test('Synthesis is deterministic, non-silent, bounded, and aligned with token duration', () => { | |
| for (const [model, layer] of [['flint',12], ['qwen',0]]) { | |
| const score = sound.buildScore(traces[model], layer, .2), first = sound.renderPCM(score), second = sound.renderPCM(score); | |
| assert.deepEqual(first, second); | |
| assert.equal(first.length, Math.ceil(score.duration * sound.SAMPLE_RATE)); | |
| let peak = 0; for (const value of first) { assert.ok(Number.isFinite(value)); peak = Math.max(peak, Math.abs(value)); } | |
| assert.ok(peak > .001 && peak < 1); | |
| assert.equal(first[0], 0); assert.equal(first.at(-1), 0); | |
| } | |
| }); | |
| test('WAV encodes actual deterministic signed PCM with the chosen volume', () => { | |
| const pcm = new Float32Array([0, .5, -.5, 1, -1]), buffer = sound.encodeWAV(pcm, .5), view = new DataView(buffer); | |
| assert.equal(Buffer.from(buffer).toString('ascii', 0, 4), 'RIFF'); | |
| assert.equal(Buffer.from(buffer).toString('ascii', 8, 12), 'WAVE'); | |
| assert.equal(view.getUint32(24,true), 22050); assert.equal(view.getUint16(22,true), 1); assert.equal(view.getUint16(34,true),16); | |
| assert.equal(view.getUint32(40,true), 10); assert.equal(buffer.byteLength,54); | |
| assert.deepEqual(Array.from({length:5}, (_,i) => view.getInt16(44+i*2,true)), [0,8192,-8192,16384,-16384]); | |
| assert.ok(new Int16Array(sound.encodeWAV(pcm,0),44).every(value => value === 0)); | |
| }); | |
| test('Reject dense Flint, full attention Qwen, excessive duration, and invalid volume', () => { | |
| assert.throws(() => sound.buildScore(traces.flint,0)); | |
| const full = traces.qwen.architecture.layer_types.indexOf('full_attention'); | |
| assert.throws(() => sound.buildScore(traces.qwen,full)); | |
| assert.throws(() => sound.buildScore(traces.flint,12,2)); | |
| assert.throws(() => sound.renderPCM({duration:91,sampleRate:22050,events:[]})); | |
| assert.throws(() => sound.encodeWAV(new Float32Array(1),-1)); | |
| }); | |
| test('ID pitch repeats every 15 IDs and Qwen is one octave above Flint', () => { | |
| assert.equal(sound.frequency(2,'flint'), sound.frequency(17,'flint')); | |
| assert.equal(sound.frequency(2,'qwen'), sound.frequency(2,'flint')*2); | |
| }); | |