Spaces:
Running
Running
File size: 3,696 Bytes
8d999b7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 | import { describe, expect, test } from "bun:test";
import {
padNumber,
formatEpisodeChunk,
formatEpisodeIndex,
formatFileIndex,
formatChunkIndex,
buildV3VideoPath,
buildV3DataPath,
buildV3EpisodesMetadataPath,
} from "@/utils/stringFormatting";
// These utilities are the foundation of v3.0 path construction.
// v2.x uses formatStringWithVars + manual padStart instead.
describe("padNumber", () => {
test("pads single digit to 3", () => {
expect(padNumber(1, 3)).toBe("001");
});
test("pads zero to 6", () => {
expect(padNumber(0, 6)).toBe("000000");
});
test("does not truncate numbers longer than length", () => {
expect(padNumber(1234, 3)).toBe("1234");
});
test("pads to exact length when already equal", () => {
expect(padNumber(42, 2)).toBe("42");
});
});
describe("formatEpisodeChunk — 3-digit padding (v2.x chunk_index, v3 chunk_index)", () => {
test("chunk 0 → '000'", () => {
expect(formatEpisodeChunk(0)).toBe("000");
});
test("chunk 1 → '001'", () => {
expect(formatEpisodeChunk(1)).toBe("001");
});
test("chunk 42 → '042'", () => {
expect(formatEpisodeChunk(42)).toBe("042");
});
test("chunk 999 → '999'", () => {
expect(formatEpisodeChunk(999)).toBe("999");
});
});
describe("formatEpisodeIndex — 6-digit padding (v2.x episode_index)", () => {
test("index 0 → '000000'", () => {
expect(formatEpisodeIndex(0)).toBe("000000");
});
test("index 42 → '000042'", () => {
expect(formatEpisodeIndex(42)).toBe("000042");
});
test("index 999999 → '999999'", () => {
expect(formatEpisodeIndex(999999)).toBe("999999");
});
});
describe("formatFileIndex — 3-digit padding (v3.0 file_index)", () => {
test("file 0 → '000'", () => {
expect(formatFileIndex(0)).toBe("000");
});
test("file 5 → '005'", () => {
expect(formatFileIndex(5)).toBe("005");
});
test("file 100 → '100'", () => {
expect(formatFileIndex(100)).toBe("100");
});
});
describe("formatChunkIndex — 3-digit padding (v3.0 chunk_index)", () => {
test("chunk 0 → '000'", () => {
expect(formatChunkIndex(0)).toBe("000");
});
test("chunk 12 → '012'", () => {
expect(formatChunkIndex(12)).toBe("012");
});
});
// v3.0 specific path builders
describe("buildV3VideoPath", () => {
test("single camera, chunk 0, file 0", () => {
expect(buildV3VideoPath("observation.image", 0, 0)).toBe(
"videos/observation.image/chunk-000/file-000.mp4",
);
});
test("nested camera key, non-zero chunk and file", () => {
expect(buildV3VideoPath("observation.images.wrist", 2, 5)).toBe(
"videos/observation.images.wrist/chunk-002/file-005.mp4",
);
});
test("two-camera SO101 dataset style", () => {
expect(buildV3VideoPath("observation.images.top", 0, 1)).toBe(
"videos/observation.images.top/chunk-000/file-001.mp4",
);
});
});
describe("buildV3DataPath", () => {
test("chunk 0, file 0", () => {
expect(buildV3DataPath(0, 0)).toBe("data/chunk-000/file-000.parquet");
});
test("chunk 1, file 3", () => {
expect(buildV3DataPath(1, 3)).toBe("data/chunk-001/file-003.parquet");
});
test("large indices", () => {
expect(buildV3DataPath(10, 99)).toBe("data/chunk-010/file-099.parquet");
});
});
describe("buildV3EpisodesMetadataPath", () => {
test("chunk 0, file 0 (default for most datasets)", () => {
expect(buildV3EpisodesMetadataPath(0, 0)).toBe(
"meta/episodes/chunk-000/file-000.parquet",
);
});
test("chunk 0, file 2 (multiple metadata files)", () => {
expect(buildV3EpisodesMetadataPath(0, 2)).toBe(
"meta/episodes/chunk-000/file-002.parquet",
);
});
});
|