// 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" ) func TestDrawYAxis(t *testing.T) { assert := assert.New(t) newDraw := func() *Draw { d, _ := NewDraw(DrawOption{ Width: 400, Height: 300, }) return d } tests := []struct { newDraw func() *Draw newOption func() *ChartOption xAxisHeight int result string }{ { newDraw: newDraw, newOption: func() *ChartOption { return &ChartOption{ YAxis: YAxisOption{ Max: toFloatPoint(20), }, SeriesList: []Series{ { Data: []SeriesData{ { Value: 1, }, { Value: 2, }, }, }, }, } }, result: "", }, } for _, tt := range tests { d := tt.newDraw() r, err := drawYAxis(d, tt.newOption(), tt.xAxisHeight, chart.NewBox(10, 10, 10, 10)) assert.Nil(err) assert.Equal(&Range{ divideCount: 6, Max: 20, Size: 280, }, r) data, err := d.Bytes() assert.Nil(err) assert.Equal(tt.result, string(data)) } }