77 lines
2.4 KiB
TypeScript
77 lines
2.4 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import {
|
|
extractByColor,
|
|
isGrayscaleish,
|
|
luma01,
|
|
rarityPredicate,
|
|
renderPredicateMask,
|
|
} from "./masks";
|
|
import { makeImage } from "./test-helpers";
|
|
|
|
const px = makeImage(2, 1, [
|
|
[255, 0, 0, 255],
|
|
[10, 10, 10, 255],
|
|
]);
|
|
|
|
describe("isGrayscaleish / luma01", () => {
|
|
it("серый распознаётся с допуском, цветной — нет", () => {
|
|
expect(isGrayscaleish(10, 10, 10, 0)).toBe(true);
|
|
expect(isGrayscaleish(12, 10, 11, 2)).toBe(true);
|
|
expect(isGrayscaleish(255, 0, 0, 0)).toBe(false);
|
|
});
|
|
|
|
it("luma01: белый 1, чёрный 0", () => {
|
|
expect(luma01(255, 255, 255)).toBeCloseTo(1);
|
|
expect(luma01(0, 0, 0)).toBeCloseTo(0);
|
|
});
|
|
});
|
|
|
|
describe("renderPredicateMask", () => {
|
|
it("binary: совпавшие белые на чёрном, альфа принудительная", () => {
|
|
const out = renderPredicateMask(px, (r) => r > 200, { mode: "binary" });
|
|
expect([...out.data.slice(0, 4)]).toEqual([255, 255, 255, 255]);
|
|
expect([...out.data.slice(4, 8)]).toEqual([0, 0, 0, 255]);
|
|
});
|
|
|
|
it("highlight: подкрашивает только совпавшие, остальные без изменений", () => {
|
|
const out = renderPredicateMask(px, (r) => r > 200, {
|
|
mode: "highlight",
|
|
color: "#0000ff",
|
|
opacityPercent: 100,
|
|
});
|
|
expect([...out.data.slice(0, 4)]).toEqual([0, 0, 255, 255]);
|
|
expect([...out.data.slice(4, 8)]).toEqual([10, 10, 10, 255]);
|
|
});
|
|
|
|
it("прозрачность оригинала сохраняется в highlight-режиме", () => {
|
|
const semi = makeImage(1, 1, [[200, 200, 200, 128]]);
|
|
const out = renderPredicateMask(semi, () => true, {
|
|
mode: "highlight",
|
|
color: "#00ff00",
|
|
opacityPercent: 50,
|
|
});
|
|
expect(out.data[3]).toBe(128);
|
|
expect(out.data[0]).toBeGreaterThan(90);
|
|
});
|
|
});
|
|
|
|
describe("rarityPredicate + extractByColor", () => {
|
|
const img = makeImage(3, 1, [
|
|
[255, 0, 0, 255],
|
|
[255, 0, 0, 255],
|
|
[0, 255, 0, 255],
|
|
]);
|
|
|
|
it("уникальный (единичный) цвет находится, массовый — нет", () => {
|
|
const pred = rarityPredicate(img, 1);
|
|
expect(pred(0, 255, 0, 255)).toBe(true);
|
|
expect(pred(255, 0, 0, 255)).toBe(false);
|
|
});
|
|
|
|
it("extractByColor оставляет близкие и делает дальние прозрачными", () => {
|
|
const out = extractByColor(px, "#0a0a0a", 5);
|
|
expect(out.data[3]).toBe(0);
|
|
expect(out.data[7]).toBe(255);
|
|
});
|
|
});
|