// MIT License
// Copyright (c) 2022 Tree Xie
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.
package charts
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/wcharczuk/go-chart/v2/drawing"
)
func TestChartOption(t *testing.T) {
assert := assert.New(t)
fns := []OptionFunc{
SVGTypeOption(),
FontFamilyOptionFunc("fontFamily"),
ThemeOptionFunc("theme"),
TitleTextOptionFunc("title"),
LegendLabelsOptionFunc([]string{
"label",
}),
XAxisDataOptionFunc([]string{
"xaxis",
}),
YAxisDataOptionFunc([]string{
"yaxis",
}),
WidthOptionFunc(800),
HeightOptionFunc(600),
PaddingOptionFunc(Box{
Left: 10,
Top: 10,
Right: 10,
Bottom: 10,
}),
BackgroundColorOptionFunc(drawing.ColorBlack),
}
opt := ChartOption{}
for _, fn := range fns {
fn(&opt)
}
assert.Equal(ChartOption{
Type: ChartOutputSVG,
FontFamily: "fontFamily",
Theme: "theme",
Title: TitleOption{
Text: "title",
},
Legend: LegendOption{
Data: []string{
"label",
},
},
XAxis: XAxisOption{
Data: []string{
"xaxis",
},
},
YAxisOptions: []YAxisOption{
{
Data: []string{
"yaxis",
},
},
},
Width: 800,
Height: 600,
Padding: Box{
Left: 10,
Top: 10,
Right: 10,
Bottom: 10,
},
BackgroundColor: drawing.ColorBlack,
}, opt)
}
func TestChartOptionPieSeriesShowLabel(t *testing.T) {
assert := assert.New(t)
opt := ChartOption{
SeriesList: NewPieSeriesList([]float64{
1,
2,
}),
}
PieSeriesShowLabel()(&opt)
assert.True(opt.SeriesList[0].Label.Show)
}
func TestChartOptionMarkLine(t *testing.T) {
assert := assert.New(t)
opt := ChartOption{
SeriesList: NewSeriesListDataFromValues([][]float64{
{1, 2},
}),
}
MarkLineOptionFunc(0, "min", "max")(&opt)
assert.Equal(NewMarkLine("min", "max"), opt.SeriesList[0].MarkLine)
}
func TestChartOptionMarkPoint(t *testing.T) {
assert := assert.New(t)
opt := ChartOption{
SeriesList: NewSeriesListDataFromValues([][]float64{
{1, 2},
}),
}
MarkPointOptionFunc(0, "min", "max")(&opt)
assert.Equal(NewMarkPoint("min", "max"), opt.SeriesList[0].MarkPoint)
}
func TestLineRender(t *testing.T) {
assert := assert.New(t)
values := [][]float64{
{
120,
132,
101,
134,
90,
230,
210,
},
{
220,
182,
191,
234,
290,
330,
310,
},
{
150,
232,
201,
154,
190,
330,
410,
},
{
320,
332,
301,
334,
390,
330,
320,
},
{
820,
932,
901,
934,
1290,
1330,
1320,
},
}
p, err := LineRender(
values,
SVGTypeOption(),
TitleTextOptionFunc("Line"),
XAxisDataOptionFunc([]string{
"Mon",
"Tue",
"Wed",
"Thu",
"Fri",
"Sat",
"Sun",
}),
LegendLabelsOptionFunc([]string{
"Email",
"Union Ads",
"Video Ads",
"Direct",
"Search Engine",
}, PositionCenter),
)
assert.Nil(err)
data, err := p.Bytes()
assert.Nil(err)
assert.Equal("", string(data))
}
func TestBarRender(t *testing.T) {
assert := assert.New(t)
values := [][]float64{
{
2.0,
4.9,
7.0,
23.2,
25.6,
76.7,
135.6,
162.2,
32.6,
20.0,
6.4,
3.3,
},
{
2.6,
5.9,
9.0,
26.4,
28.7,
70.7,
175.6,
182.2,
48.7,
18.8,
6.0,
2.3,
},
}
p, err := BarRender(
values,
SVGTypeOption(),
XAxisDataOptionFunc([]string{
"Jan",
"Feb",
"Mar",
"Apr",
"May",
"Jun",
"Jul",
"Aug",
"Sep",
"Oct",
"Nov",
"Dec",
}),
LegendLabelsOptionFunc([]string{
"Rainfall",
"Evaporation",
}, PositionRight),
MarkLineOptionFunc(0, SeriesMarkDataTypeAverage),
MarkPointOptionFunc(0, SeriesMarkDataTypeMax,
SeriesMarkDataTypeMin),
// custom option func
func(opt *ChartOption) {
opt.SeriesList[1].MarkPoint = NewMarkPoint(
SeriesMarkDataTypeMax,
SeriesMarkDataTypeMin,
)
opt.SeriesList[1].MarkLine = NewMarkLine(
SeriesMarkDataTypeAverage,
)
},
)
assert.Nil(err)
data, err := p.Bytes()
assert.Nil(err)
assert.Equal("", string(data))
}
func TestHorizontalBarRender(t *testing.T) {
assert := assert.New(t)
values := [][]float64{
{
18203,
23489,
29034,
104970,
131744,
630230,
},
{
19325,
23438,
31000,
121594,
134141,
681807,
},
}
p, err := HorizontalBarRender(
values,
SVGTypeOption(),
TitleTextOptionFunc("World Population"),
PaddingOptionFunc(Box{
Top: 20,
Right: 40,
Bottom: 20,
Left: 20,
}),
LegendLabelsOptionFunc([]string{
"2011",
"2012",
}),
YAxisDataOptionFunc([]string{
"Brazil",
"Indonesia",
"USA",
"India",
"China",
"World",
}),
)
assert.Nil(err)
data, err := p.Bytes()
assert.Nil(err)
assert.Equal("", string(data))
}
func TestPieRender(t *testing.T) {
assert := assert.New(t)
values := []float64{
1048,
735,
580,
484,
300,
}
p, err := PieRender(
values,
SVGTypeOption(),
TitleOptionFunc(TitleOption{
Text: "Rainfall vs Evaporation",
Subtext: "Fake Data",
Left: PositionCenter,
}),
PaddingOptionFunc(Box{
Top: 20,
Right: 20,
Bottom: 20,
Left: 20,
}),
LegendOptionFunc(LegendOption{
Orient: OrientVertical,
Data: []string{
"Search Engine",
"Direct",
"Email",
"Union Ads",
"Video Ads",
},
Left: PositionLeft,
}),
PieSeriesShowLabel(),
)
assert.Nil(err)
data, err := p.Bytes()
assert.Nil(err)
assert.Equal("", string(data))
}
func TestRadarRender(t *testing.T) {
assert := assert.New(t)
values := [][]float64{
{
4200,
3000,
20000,
35000,
50000,
18000,
},
{
5000,
14000,
28000,
26000,
42000,
21000,
},
}
p, err := RadarRender(
values,
SVGTypeOption(),
TitleTextOptionFunc("Basic Radar Chart"),
LegendLabelsOptionFunc([]string{
"Allocated Budget",
"Actual Spending",
}),
RadarIndicatorOptionFunc([]string{
"Sales",
"Administration",
"Information Technology",
"Customer Support",
"Development",
"Marketing",
}, []float64{
6500,
16000,
30000,
38000,
52000,
25000,
}),
)
assert.Nil(err)
data, err := p.Bytes()
assert.Nil(err)
assert.Equal("", string(data))
}
func TestFunnelRender(t *testing.T) {
assert := assert.New(t)
values := []float64{
100,
80,
60,
40,
20,
}
p, err := FunnelRender(
values,
SVGTypeOption(),
TitleTextOptionFunc("Funnel"),
LegendLabelsOptionFunc([]string{
"Show",
"Click",
"Visit",
"Inquiry",
"Order",
}),
)
assert.Nil(err)
data, err := p.Bytes()
assert.Nil(err)
assert.Equal("", string(data))
}