File size: 1,149 Bytes
21dd449 |
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 |
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-dev-test") {
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).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),
},
]);
});
});
|