homeclaw/tests/gpu-specs.test.ts

/**
 * gpu-specs — 모델명 인식과 카드 선택
 *
 * 순수 함수만 테스트한다. 다운로드/캐시는 네트워크와 디스크에 의존하므로 여기서 다루지 않는다.
 *
 * 이 테스트가 지키는 두 가지:
 *   1) 맨숫자를 언제 모델명으로 볼 것인가 — "4080이 5060보다 빠른가"는 잡아야 하지만
 *      "2024년 실적"과 "1080p 영상"은 잡으면 안 된다. 무관한 답변에 GPU 스펙이 끼어들면
 *      그 자체가 오염이다.
 *   2) 변형 모델 선택 — "RTX 4080"은 SUPER/Mobile 이름에도 부분일치하는데, 데스크탑 질문에
 *      노트북 칩 수치를 물려주면 조용히 틀린 답이 된다.
 */

import { test, describe } from 'node:test';
import assert from 'node:assert/strict';
import { extractGpuMentions, findGpu, formatGpuRecord, type GpuRecord } from '../src/tools/gpu-specs';

const GPUS: GpuRecord[] = [
  { name: 'GeForce RTX 4080', vendor: 'nvidia', releaseDate: '2022-09-20', architecture: 'Ada Lovelace', processSize: 5, shaders: 9728, tensorCores: 304, baseClock: 2205, boostClock: 2505, memorySize: 16, memoryType: 'GDDR6X', memoryBus: 256, memoryBandwidth: 716.8, fp32: 48.74, tdp: 320, url: 'https://example.invalid/4080' },
  { name: 'GeForce RTX 4080 SUPER', vendor: 'nvidia', releaseDate: '2024-01-31' },
  { name: 'GeForce RTX 4080 Mobile', vendor: 'nvidia', releaseDate: '2023-02-22' },
  { name: 'GeForce RTX 5060', vendor: 'nvidia', releaseDate: '2025-05-19', memoryBandwidth: 448, fp32: 19.18 },
  { name: 'GeForce RTX 5060 Ti 16 GB', vendor: 'nvidia', releaseDate: '2025-04-16' },
  { name: 'GeForce RTX 5060 Mobile', vendor: 'nvidia', releaseDate: '2025-05-01' },
  { name: 'Radeon RX 7900 XTX', vendor: 'amd', releaseDate: '2022-12-13' },
  { name: 'Radeon RX 7900 XT', vendor: 'amd', releaseDate: '2022-12-13' },
];

describe('extractGpuMentions — 모델명 인식', () => {
  test('시리즈 표기가 붙은 형태', () => {
    const m = extractGpuMentions('RTX 4080 vs RTX 5060 performance comparison specs');
    assert.deepEqual(m.map(x => `${x.series} ${x.number}`), ['rtx 4080', 'rtx 5060']);
  });

  test('접미사(Ti/SUPER/XTX)를 보존한다', () => {
    assert.equal(extractGpuMentions('RTX 5060 Ti 리뷰')[0].suffix, 'ti');
    assert.equal(extractGpuMentions('RX 7900 XTX 성능')[0].suffix, 'xtx');
  });

  test('GPU 맥락이 있으면 맨숫자도 모델명으로 본다', () => {
    // 사용자가 실제로 친 형태. "RTX"가 한 번만 나오고 나머지는 숫자만 쓴다.
    const m = extractGpuMentions('RTX 4080이 5060 보다 얼마나 빠르지?');
    assert.deepEqual(m.map(x => x.number).sort(), ['4080', '5060']);
  });

  test('GPU 맥락이 없으면 맨숫자를 건드리지 않는다', () => {
    assert.deepEqual(extractGpuMentions('2024년 매출이 3800억이었다'), []);
    assert.deepEqual(extractGpuMentions('아파트 3800세대 분양'), []);
  });

  test('연도 꼴은 맥락이 있어도 모델명으로 보지 않는다', () => {
    const m = extractGpuMentions('이 GPU는 2022년에 나왔다');
    assert.deepEqual(m, []);
  });

  test('중복은 한 번만', () => {
    const m = extractGpuMentions('RTX 4080 성능, RTX 4080 가격');
    assert.equal(m.length, 1);
  });
});

describe('findGpu — 변형 모델에 밀리지 않고 기본 카드를 고른다', () => {
  const pick = (q: string) => findGpu(extractGpuMentions(q)[0], GPUS)?.name;

  test('"RTX 4080"은 SUPER나 Mobile이 아니라 기본 카드', () => {
    assert.equal(pick('RTX 4080 specs'), 'GeForce RTX 4080');
  });

  test('"RTX 5060"은 Ti나 Mobile이 아니라 기본 카드', () => {
    assert.equal(pick('RTX 5060 specs'), 'GeForce RTX 5060');
  });

  test('접미사를 명시하면 그 변형을 고른다', () => {
    assert.equal(pick('RTX 4080 SUPER 스펙'), 'GeForce RTX 4080 SUPER');
    assert.equal(pick('RTX 5060 Ti 스펙'), 'GeForce RTX 5060 Ti 16 GB');
  });

  test('XT와 XTX를 구분한다', () => {
    assert.equal(pick('RX 7900 XTX 성능'), 'Radeon RX 7900 XTX');
    assert.equal(pick('RX 7900 XT 성능'), 'Radeon RX 7900 XT');
  });

  test('데이터에 없는 모델은 null', () => {
    assert.equal(findGpu({ series: 'rtx', number: '9090', suffix: '', raw: 'RTX 9090' }, GPUS), null);
  });
});

describe('formatGpuRecord — 판단에 필요한 필드만', () => {
  const out = formatGpuRecord(GPUS[0]);

  test('출시일이 들어간다 — "미출시" 환각을 원천 차단하는 필드다', () => {
    assert.match(out, /2022-09-20/);
  });

  test('비교에 쓰이는 수치가 들어간다', () => {
    assert.match(out, /716\.8 GB\/s/);
    assert.match(out, /FP32 48\.74 TFLOPS/);
    assert.match(out, /9728/);
  });

  test('없는 필드는 빈 줄로 남기지 않는다', () => {
    const sparse = formatGpuRecord({ name: 'GeForce RTX 5060', releaseDate: '2025-05-19' });
    assert.equal(sparse.split('\n').some(l => l.trim() === ''), false);
    assert.match(sparse, /2025-05-19/);
  });
});