Spaces:
Running
Running
| const fs = require("fs"); | |
| const path = require("path"); | |
| const { extractFromHTML } = require("../../data_extraction/extractor"); | |
| describe("extractor", () => { | |
| const sampleFile = path.join(__dirname, "sample.html"); | |
| beforeAll(() => { | |
| fs.writeFileSync( | |
| sampleFile, | |
| ` | |
| <h1>Q: What is service?</h1> | |
| <p>A: Service is love made visible.</p> | |
| ` | |
| ); | |
| }); | |
| afterAll(() => { | |
| fs.unlinkSync(sampleFile); | |
| }); | |
| test("extracts paragraphs and roles", () => { | |
| const turns = extractFromHTML(sampleFile); | |
| expect(turns.length).toBe(2); | |
| expect(turns[0].role).toBe("user"); | |
| expect(turns[0].content).toContain("What is service"); | |
| expect(turns[1].role).toBe("assistant"); | |
| expect(turns[1].content).toContain("love made visible"); | |
| }); | |
| }); | |