weijuesen
22 lines
678 B
TypeScript
22 lines
678 B
TypeScript
import { describe, expect, it } from 'vitest';
|
|
import { aiClassLabel, riskLevelLabel } from './format';
|
|
|
|
describe('aiClassLabel', () => {
|
|
it('映射 healthy/sick 为中文', () => {
|
|
expect(aiClassLabel('healthy')).toBe('健康');
|
|
expect(aiClassLabel('sick')).toBe('疑似异常');
|
|
});
|
|
it('未知类别原样返回', () => {
|
|
expect(aiClassLabel('unknown')).toBe('unknown');
|
|
});
|
|
});
|
|
|
|
describe('riskLevelLabel', () => {
|
|
it('映射四级风险', () => {
|
|
expect(riskLevelLabel('green')).toBe('绿');
|
|
expect(riskLevelLabel('yellow')).toBe('黄');
|
|
expect(riskLevelLabel('orange')).toBe('橙');
|
|
expect(riskLevelLabel('red')).toBe('红');
|
|
});
|
|
});
|