import { Layout, type PageFragment } from '$gql/graphql'; import { partition, type Chunk } from '$lib/Reader'; import { expect, test } from 'vitest'; const normalAttrs = { aspectRatio: 0.7, width: 140, height: 200 }; const wideAttrs = { aspectRatio: 1.4, width: 280, height: 200 }; const pages: PageFragment[] = [ { id: 0, path: '000.png', image: { id: 0, hash: '0', ...normalAttrs } }, { id: 1, path: '001.png', image: { id: 1, hash: '1', ...normalAttrs } }, { id: 2, path: '002.png', image: { id: 2, hash: '2', ...normalAttrs } }, { id: 3, path: '003.png', image: { id: 3, hash: '3', ...normalAttrs } }, { id: 4, path: '004.png', image: { id: 4, hash: '4', ...wideAttrs } }, { id: 5, path: '005.png', image: { id: 5, hash: '5', ...normalAttrs } }, { id: 6, path: '006.png', image: { id: 6, hash: '6', ...normalAttrs } } ]; const ids = (chunks: Chunk[]) => chunks.map((c) => (c.secondary ? [c.main.id, c.secondary.id] : [c.main.id])); const indices = (chunks: Chunk[]) => chunks.map((c) => c.index); test('partitions single layout', () => { const [chunks, lookup] = partition(pages, Layout.Single); expect(ids(chunks)).toStrictEqual([[0], [1], [2], [3], [4], [5], [6]]); expect(indices(chunks)).toStrictEqual([0, 1, 2, 3, 4, 5, 6]); expect(lookup).toStrictEqual([0, 1, 2, 3, 4, 5, 6]); }); test('partitions double layout', () => { const [chunks, lookup] = partition(pages, Layout.Double); expect(ids(chunks)).toStrictEqual([[0, 1], [2, 3], [4], [5, 6]]); expect(indices(chunks)).toStrictEqual([0, 2, 4, 5]); expect(lookup).toStrictEqual([0, 0, 1, 1, 2, 3, 3]); }); test('partitions double (offset) layout', () => { const [chunks, lookup] = partition(pages, Layout.DoubleOffset); expect(ids(chunks)).toStrictEqual([[0], [1, 2], [3], [4], [5, 6]]); expect(indices(chunks)).toStrictEqual([0, 1, 3, 4, 5]); expect(lookup).toStrictEqual([0, 1, 1, 2, 3, 4, 4]); });