WFA-JS/tests/test.js

import wf_align from "../src/wfa.js";
import fs from "fs";
import ProgressBar from "progress";

let data = fs.readFileSync("./tests/tests.json");
data = JSON.parse(data);
const sequences = fs.readFileSync("./tests/sequences").toString().split("\n");
// const total = sequences.length;
const total = 500; // skip the later tests because of memory usage
const timePerChar = [];

for (const test_name of Object.keys(data)) {
	const test = data[test_name];
	const penalties = test.penalties;
	const solutions = fs.readFileSync(test.solutions).toString().split("\n");
	const bar = new ProgressBar(":bar :current/:total", { total: total / 2 });
	console.log(`test: ${test_name}`);
	let correct = 0;
	let j = 0;
	for (let i = 0; i < total; i += 2) {
		const s1 = sequences[i].replace(">");
		const s2 = sequences[i + 1].replace("<");
		const start = process.hrtime()[1];
		const { score } = wf_align(s1, s2, penalties, false);
		const elapsed = process.hrtime()[1] - start;
		timePerChar.push((elapsed / 1e9) / (s1.length + s2.length));
		const solution_score = Number(solutions[j].split("\t")[0]);
		if (solution_score === -score) {
			correct += 1;
		}
		j += 1;
		bar.tick();
	}
	console.log(`correct: ${correct}\ntotal: ${total / 2}\n`);
	console.log(`average time per character (ms): ${average(timePerChar) * 1000}`);
}

function average (arr) {
	const sum = arr.reduce((a, b) => a + b, 0);
	return sum / arr.length;
}