GPT-Vis/__tests__/radar.test.ts at 36385b7cd37712e793f6ee55ee8345a21d3b6fdc · antvis/GPT-Vis · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
import { describe, expect, it } from 'vitest';
import { parse } from '../src/syntax/parser';

describe('parse - radar chart', () => {
  it('should parse basic radar chart', () => {
    const result = parse(`
vis radar
data
  - name 沟通能力
    value 2
  - name 协作能力
    value 3
  - name 领导能力
    value 2
  - name 学习能力
    value 5
  - name 创新能力
    value 6
  - name 技术能力
    value 9
    `);

    expect(result.type).toBe('radar');
    expect(result.data).toEqual([
      { name: '沟通能力', value: 2 },
      { name: '协作能力', value: 3 },
      { name: '领导能力', value: 2 },
      { name: '学习能力', value: 5 },
      { name: '创新能力', value: 6 },
      { name: '技术能力', value: 9 },
    ]);
  });

  it('should parse grouped radar chart', () => {
    const result = parse(`
vis radar
data
  - name 语文
    value 95
    group 一班
  - name 数学
    value 96
    group 一班
  - name 外语
    value 85
    group 一班
  - name 语文
    value 75
    group 二班
  - name 数学
    value 93
    group 二班
  - name 外语
    value 66
    group 二班
    `);

    expect(result.type).toBe('radar');
    expect(result.data).toEqual([
      { name: '语文', value: 95, group: '一班' },
      { name: '数学', value: 96, group: '一班' },
      { name: '外语', value: 85, group: '一班' },
      { name: '语文', value: 75, group: '二班' },
      { name: '数学', value: 93, group: '二班' },
      { name: '外语', value: 66, group: '二班' },
    ]);
  });

  it('should parse radar chart with title and theme', () => {
    const result = parse(`
vis radar
data
  - name Vitamin C
    value 7
  - name Fiber
    value 6
  - name Sugar
    value 5
title 营养成分分析
theme academy
    `);

    expect(result.title).toBe('营养成分分析');
    expect(result.theme).toBe('academy');
  });
});