// 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"
)
func TestLineChart(t *testing.T) {
assert := assert.New(t)
tests := []struct {
render func(*Painter) ([]byte, error)
result string
}{
{
render: func(p *Painter) ([]byte, error) {
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,
},
}
_, err := NewLineChart(p, LineChartOption{
Title: TitleOption{
Text: "Line",
},
Padding: Box{
Top: 10,
Right: 10,
Bottom: 10,
Left: 10,
},
XAxis: NewXAxisOption([]string{
"Mon",
"Tue",
"Wed",
"Thu",
"Fri",
"Sat",
"Sun",
}),
Legend: NewLegendOption([]string{
"Email",
"Union Ads",
"Video Ads",
"Direct",
"Search Engine",
}, PositionCenter),
SeriesList: NewSeriesListDataFromValues(values),
}).Render()
if err != nil {
return nil, err
}
return p.Bytes()
},
result: "",
},
{
render: func(p *Painter) ([]byte, error) {
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,
},
}
_, err := NewLineChart(p, LineChartOption{
Title: TitleOption{
Text: "Line",
},
Padding: Box{
Top: 10,
Right: 10,
Bottom: 10,
Left: 10,
},
XAxis: NewXAxisOption([]string{
"Mon",
"Tue",
"Wed",
"Thu",
"Fri",
"Sat",
"Sun",
}, FalseFlag()),
Legend: NewLegendOption([]string{
"Email",
"Union Ads",
"Video Ads",
"Direct",
"Search Engine",
}, PositionCenter),
SeriesList: NewSeriesListDataFromValues(values),
}).Render()
if err != nil {
return nil, err
}
return p.Bytes()
},
result: "",
},
{
render: func(p *Painter) ([]byte, error) {
values := [][]float64{
{
1200,
1320,
1010,
1340,
900,
2300,
2100,
},
{
2200,
1820,
1910,
2340,
2900,
3300,
3100,
},
{
1500,
2320,
2010,
1540,
1900,
3300,
4100,
},
{
3200,
3320,
3010,
3340,
3900,
3300,
3200,
},
{
8200,
9320,
9010,
9340,
1290,
1330,
1320,
},
}
_, err := NewLineChart(p, LineChartOption{
Title: TitleOption{
Text: "Line",
},
Padding: Box{
Top: 10,
Right: 10,
Bottom: 10,
Left: 10,
},
XAxis: NewXAxisOption([]string{
"Mon",
"Tue",
"Wed",
"Thu",
"Fri",
"Sat",
"Sun",
}, FalseFlag()),
Legend: NewLegendOption([]string{
"Email",
"Union Ads",
"Video Ads",
"Direct",
"Search Engine",
}, PositionCenter),
SeriesList: func(list SeriesList) SeriesList {
for index := range list {
list[index].Label.Show = true
list[index].Label.Position = "top"
list[index].Label.Formatter = "{e}"
}
return list
}(NewSeriesListDataFromValues(values)),
}).Render()
if err != nil {
return nil, err
}
return p.Bytes()
},
result: "",
},
}
for _, tt := range tests {
p, err := NewPainter(PainterOptions{
Type: ChartOutputSVG,
Width: 600,
Height: 400,
}, PainterThemeOption(defaultTheme))
assert.Nil(err)
data, err := tt.render(p)
assert.Nil(err)
assert.Equal(tt.result, string(data))
}
}