@huggingface/hub/src/lib/list-datasets.spec.ts

Utilities to interact with the Hugging Face hub

import { describe, expect, it } from "vitest";
import type { DatasetEntry } from "./list-datasets";
import { listDatasets } from "./list-datasets";

describe("listDatasets", () => {
	it("should list datasets from hf-doc-builder", async () => {
		const results: DatasetEntry[] = [];

		for await (const entry of listDatasets({ search: { owner: "hf-doc-build" } })) {
			if (entry.name !== "hf-doc-build/doc-build" && entry.name !== "hf-doc-build/doc-build-dev") {
				continue;
			}
			if (typeof entry.downloads === "number") {
				entry.downloads = 0;
			}
			if (typeof entry.likes === "number") {
				entry.likes = 0;
			}
			if (entry.updatedAt instanceof Date && !isNaN(entry.updatedAt.getTime())) {
				entry.updatedAt = new Date(0);
			}

			results.push(entry);
		}

		expect(results.sort((a, b) => a.id.localeCompare(b.id))).to.deep.equal([
			{
				id: "6356b19985da6f13863228bd",
				name: "hf-doc-build/doc-build",
				private: false,
				gated: false,
				downloads: 0,
				likes: 0,
				updatedAt: new Date(0),
			},
			{
				id: "636a1b69f2f9ec4289c4c19e",
				name: "hf-doc-build/doc-build-dev",
				gated: false,
				private: false,
				downloads: 0,
				likes: 0,
				updatedAt: new Date(0),
			},
		]);
	});
});