add protein

2 files changed, 39 insertions, 0 deletions

diff --git a/src/protein/protein.js b/src/protein/protein.js
new file mode 100644
index 0000000..0223203
--- /dev/null
+++ b/src/protein/protein.js
@@ -0,0 +1,10 @@
+const xnor = (a, b) => a === b ? 1 : 0;
+
+export function compare(pa, pb) {
+	if (pa.length !== pb.length) {
+		throw new Error(`attempted to compare proteins with different lengths: ${pa.length} vs ${pb.length}`);
+	}
+	return (1/pa.length) * pa
+		.map((_, i) => xnor(pa[i], pb[i]))
+		.reduce((acc, val) => acc + val, 0)
+}
diff --git a/src/protein/protein.test.js b/src/protein/protein.test.js
new file mode 100644
index 0000000..61c4896
--- /dev/null
+++ b/src/protein/protein.test.js
@@ -0,0 +1,29 @@
+'use strict';
+
+import { compare } from './protein';
+
+test('compare size-1 proteins', () => {
+	expect(compare([0], [0])).toBe(1);
+	expect(compare([0], [1])).toBe(0);
+	expect(compare([1], [0])).toBe(0);
+	expect(compare([1], [1])).toBe(1);
+});
+
+
+test('compare larger proteins', () => {
+	expect(compare([0,0], [0,0])).toBe(1);
+	expect(compare([1,1], [1,1])).toBe(1);
+	expect(compare([0,1], [1,1])).toBeCloseTo(0.5);
+	expect(compare([0,1,0], [1,1,1])).toBeCloseTo(1/3);
+	expect(compare([0,1,0,1], [1,1,1,1])).toBeCloseTo(0.5);
+	expect(compare([0,1,0,1], [0,1,1,1])).toBeCloseTo(0.75);
+	expect(compare([0,1,0,1], [0,1,1,0])).toBeCloseTo(0.5);
+	expect(compare([0,1,0,0], [0,1,1,0])).toBeCloseTo(0.75);
+	expect(compare([0,1,1,0], [0,1,1,0])).toBeCloseTo(1);
+});
+
+
+test('comparisons between differently-sized proteins throw', () => {
+	expect(() => compare([0], [0, 0])).toThrow();
+	expect(() => compare([0, 0], [0])).toThrow();
+});