52 lines
2.7 KiB
TypeScript
52 lines
2.7 KiB
TypeScript
import { expect, test } from "bun:test";
|
|
import { scoreSong, type PipelineOutput } from "../../src/eval/score.ts";
|
|
import { secondsToTicks } from "../../src/core/time.ts";
|
|
import type { TimedNote, ScoreNote, TruthTrack } from "../../src/core/types.ts";
|
|
|
|
const flags = { oracleStems: false, oracleTempo: false, oracleNotes: false };
|
|
const bpm = 120; // 0.5 s per beat
|
|
const n = (onset: number, pitch: number, string?: number): TimedNote => ({ onset, offset: onset + 0.5, pitch, string });
|
|
const sc = (t: TimedNote[]): ScoreNote[] => t.map((x) => ({
|
|
tick: Math.round(secondsToTicks(x.onset, bpm)), dur: Math.round(secondsToTicks(x.offset - x.onset, bpm)), pitch: x.pitch, string: x.string,
|
|
}));
|
|
const grid = { bpm, firstDownbeat: 0, beatsPerBar: 4 };
|
|
|
|
test("two ref guitars, one merged out track", () => {
|
|
const g1 = [n(0, 64, 1), n(0.5, 62, 2)];
|
|
const g2 = [n(0, 40, 6), n(0.5, 45, 5)];
|
|
const truth = { bpm, timeSig: [4, 4] as [number, number], tracks: [
|
|
{ name: "g1", role: "guitar", notes: g1 }, { name: "g2", role: "guitar", notes: g2 } ] as TruthTrack[] };
|
|
const merged = [...g1, ...g2];
|
|
const out: PipelineOutput = { grid, tracks: [{ name: "o", role: "guitar", notes: merged, scoreNotes: sc(merged) }] };
|
|
const s = scoreSong(truth, out, flags);
|
|
expect(s.guitarMerged.f1).toBe(1);
|
|
expect(s.guitarMerged.pass).toBe(true);
|
|
expect(s.partSplit).not.toBeNull();
|
|
expect(s.partSplit!.outTracks).toBe(1);
|
|
expect(s.partSplit!.refTracks).toBe(2);
|
|
expect(s.partSplit!.f1).toBeCloseTo(0.5, 6);
|
|
expect(s.rhythm).toEqual({ factor: 1, beatOffset: 0, onBeat: 1, durationOk: 1 });
|
|
expect(s.stringAgreement).toBe(1);
|
|
expect(s.spuriousRate).toBe(0);
|
|
expect(s.perRole.vocals!.f1).toBeNull();
|
|
expect(s.drumsPerClass).toBeNull();
|
|
});
|
|
test("partSplit null with one ref guitar; rhythm null without guitar", () => {
|
|
const truth = { bpm, timeSig: [4, 4] as [number, number], tracks: [{ name: "g", role: "guitar", notes: [n(0, 60)] }] as TruthTrack[] };
|
|
const s = scoreSong(truth, { grid, tracks: [] }, flags);
|
|
expect(s.partSplit).toBeNull();
|
|
expect(s.guitarMerged.recall).toBe(0);
|
|
expect(s.guitarMerged.pass).toBe(false);
|
|
expect(s.rhythm).toBeNull();
|
|
expect(s.spuriousRate).toBeNull();
|
|
});
|
|
test("drums per class", () => {
|
|
const truth = { bpm, timeSig: [4, 4] as [number, number], tracks: [{ name: "d", role: "drums", notes: [n(0, 36), n(0.5, 38)] }] as TruthTrack[] };
|
|
const est = [n(0, 35), n(0.5, 42)];
|
|
const s = scoreSong(truth, { grid, tracks: [{ name: "d", role: "drums", notes: est, scoreNotes: [] }] }, flags);
|
|
expect(s.drumsPerClass!.kick.f1).toBe(1);
|
|
expect(s.drumsPerClass!.snare.recall).toBe(0);
|
|
expect(s.drumsPerClass!.hihat.precision).toBe(0);
|
|
expect(s.drumsPerClass!.tom.f1).toBeNull();
|
|
});
|