38 lines
1.1 KiB
TypeScript
38 lines
1.1 KiB
TypeScript
import { describe, it, expect } from "vitest";
|
|
import { SentimentAnalyst } from "../sentiment";
|
|
import type { OpenRouterClient } from "../../lib/openrouter";
|
|
|
|
describe("SentimentAnalyst", () => {
|
|
it("should analyze sentiment from headlines", async () => {
|
|
const mockClient = {
|
|
createChatCompletion: async () => ({
|
|
choices: [
|
|
{
|
|
message: {
|
|
content:
|
|
'{"signal":"bullish","confidence":0.85,"reasoning":"Positive sentiment from news"}',
|
|
},
|
|
},
|
|
],
|
|
}),
|
|
} as unknown as OpenRouterClient;
|
|
|
|
const analyst = new SentimentAnalyst(mockClient);
|
|
const result = await analyst.analyze("AAPL", {
|
|
headlines: ["Apple beats earnings", "New iPhone launch"],
|
|
source: "news",
|
|
});
|
|
|
|
expect(result.analyst).toBe("sentiment");
|
|
expect(result.signal.signal).toBe("bullish");
|
|
});
|
|
|
|
it("should use specified model", () => {
|
|
const mockClient = {} as unknown as OpenRouterClient;
|
|
const analyst = new SentimentAnalyst(mockClient, {
|
|
model: "custom/model",
|
|
});
|
|
|
|
expect(analyst.getModel()).toBe("custom/model");
|
|
});
|
|
}); |