35 lines
1.1 KiB
TypeScript
35 lines
1.1 KiB
TypeScript
import { describe, it, expect } from 'vitest';
|
|
import { buildColorScale, FALLBACK_COLORS } from '../src/shared/color-scale';
|
|
import type { DayEntry } from '../src/types';
|
|
|
|
describe('buildColorScale', () => {
|
|
const days: DayEntry[] = [
|
|
{ date: '2025-01-01', hours: 2, count: 3 },
|
|
{ date: '2025-01-02', hours: 8, count: 5 },
|
|
{ date: '2025-01-03', hours: 0, count: 0 },
|
|
];
|
|
|
|
it('builds scale for hours metric', () => {
|
|
const scale = buildColorScale(days, 'hours');
|
|
// max hours is 8, so domain is [0, 8]
|
|
expect(scale(0)).toBe(FALLBACK_COLORS[0]);
|
|
expect(scale(8)).toBe(FALLBACK_COLORS[3]);
|
|
});
|
|
|
|
it('builds scale for count metric', () => {
|
|
const scale = buildColorScale(days, 'count');
|
|
// max count is 5, so domain is [0, 5]
|
|
expect(scale(0)).toBe(FALLBACK_COLORS[0]);
|
|
expect(scale(5)).toBe(FALLBACK_COLORS[3]);
|
|
});
|
|
|
|
it('defaults to hours metric', () => {
|
|
const scale = buildColorScale(days);
|
|
expect(scale(8)).toBe(FALLBACK_COLORS[3]);
|
|
});
|
|
|
|
it('handles empty array with domain [0, 1]', () => {
|
|
const scale = buildColorScale([]);
|
|
expect(scale.domain()).toEqual([0, 1]);
|
|
});
|
|
});
|