mirror of
https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl
synced 2025-08-06 06:36:28 +08:00
added unit tests
This commit is contained in:
parent
5c81ea1803
commit
eab30c474b
@ -0,0 +1,33 @@
|
||||
import { removeDefaultProperty } from "./llmExtract";
|
||||
|
||||
describe("removeDefaultProperty", () => {
|
||||
it("should remove the default property from a simple object", () => {
|
||||
const input = { default: "test", test: "test" };
|
||||
const expectedOutput = { test: "test" };
|
||||
expect(removeDefaultProperty(input)).toEqual(expectedOutput);
|
||||
});
|
||||
|
||||
it("should remove the default property from a nested object", () => {
|
||||
const input = { default: "test", nested: { default: "nestedTest", test: "nestedTest" } };
|
||||
const expectedOutput = { nested: { test: "nestedTest" } };
|
||||
expect(removeDefaultProperty(input)).toEqual(expectedOutput);
|
||||
});
|
||||
|
||||
it("should remove the default property from an array of objects", () => {
|
||||
const input = { array: [{ default: "test1", test: "test1" }, { default: "test2", test: "test2" }] };
|
||||
const expectedOutput = { array: [{ test: "test1" }, { test: "test2" }] };
|
||||
expect(removeDefaultProperty(input)).toEqual(expectedOutput);
|
||||
});
|
||||
|
||||
it("should handle objects without a default property", () => {
|
||||
const input = { test: "test" };
|
||||
const expectedOutput = { test: "test" };
|
||||
expect(removeDefaultProperty(input)).toEqual(expectedOutput);
|
||||
});
|
||||
|
||||
it("should handle null and non-object inputs", () => {
|
||||
expect(removeDefaultProperty(null)).toBeNull();
|
||||
expect(removeDefaultProperty("string")).toBe("string");
|
||||
expect(removeDefaultProperty(123)).toBe(123);
|
||||
});
|
||||
});
|
@ -199,7 +199,7 @@ export async function performLLMExtract(meta: Meta, document: Document): Promise
|
||||
return document;
|
||||
}
|
||||
|
||||
function removeDefaultProperty(schema: any): any {
|
||||
export function removeDefaultProperty(schema: any): any {
|
||||
if (typeof schema !== 'object' || schema === null) return schema;
|
||||
|
||||
const { default: _, ...rest } = schema;
|
||||
|
Loading…
x
Reference in New Issue
Block a user