go-chart/annotation_series_test.go

package chart

import (
	"image/color"
	"testing"
	"time"

	"github.com/blendlabs/go-assert"
	"github.com/wcharczuk/go-chart/drawing"
)

func TestCreateContinuousSeriesLastValueLabel(t *testing.T) {
	assert := assert.New(t)

	as := CreateContinuousSeriesLastValueLabel("test", []float64{1, 2, 3}, []float64{4, 5, 6}, FloatValueFormatter)
	assert.Equal("test", as.GetName())
	assert.False(as.GetStyle().IsZero())
	assert.True(as.GetStyle().Show)
	assert.NotEmpty(as.Annotations)
	assert.Equal(3.0, as.Annotations[0].X)
	assert.Equal(6.0, as.Annotations[0].Y)
	assert.Equal("6.00", as.Annotations[0].Label)
}

func TestCreateTimeSeriesLastValueLabel(t *testing.T) {
	assert := assert.New(t)

	d := time.Now().AddDate(0, 0, -1)

	as := CreateTimeSeriesLastValueLabel("test", []time.Time{
		time.Now().AddDate(0, 0, -3),
		time.Now().AddDate(0, 0, -2),
		d,
	}, []float64{4, 5, 6}, FloatValueFormatter)
	assert.Equal("test", as.GetName())
	assert.False(as.GetStyle().IsZero())
	assert.True(as.GetStyle().Show)
	assert.NotEmpty(as.Annotations)
	assert.Equal(float64(d.Unix()), as.Annotations[0].X)
	assert.Equal(6.0, as.Annotations[0].Y)
	assert.Equal("6.00", as.Annotations[0].Label)
}

func TestAnnotationSeriesMeasure(t *testing.T) {
	assert := assert.New(t)

	as := AnnotationSeries{
		Style: Style{
			Show: true,
		},
		Annotations: []Annotation{
			Annotation{X: 1.0, Y: 1.0, Label: "1.0"},
			Annotation{X: 2.0, Y: 2.0, Label: "2.0"},
			Annotation{X: 3.0, Y: 3.0, Label: "3.0"},
			Annotation{X: 4.0, Y: 4.0, Label: "4.0"},
		},
	}

	r, err := PNG(110, 110)
	assert.Nil(err)

	f, err := GetDefaultFont()
	assert.Nil(err)

	xrange := Range{
		Min:    1.0,
		Max:    4.0,
		Domain: 100,
	}
	yrange := Range{
		Min:    1.0,
		Max:    4.0,
		Domain: 100,
	}

	cb := Box{
		Top:    5,
		Left:   5,
		Right:  105,
		Bottom: 105,
		Height: 100,
		Width:  100,
	}
	sd := Style{
		FontSize: 10.0,
		Font:     f,
	}

	box := as.Measure(r, cb, xrange, yrange, sd)
	assert.False(box.IsZero())
	assert.Equal(-5.0, box.Top)
	assert.Equal(5.0, box.Left)
	assert.Equal(139.0, box.Right) //the top,left annotation sticks up 5px and out ~44px.
	assert.Equal(115.0, box.Bottom)
}

func TestAnnotationSeriesRender(t *testing.T) {
	assert := assert.New(t)

	as := AnnotationSeries{
		Style: Style{
			Show:        true,
			FillColor:   drawing.ColorWhite,
			StrokeColor: drawing.ColorBlack,
		},
		Annotations: []Annotation{
			Annotation{X: 1.0, Y: 1.0, Label: "1.0"},
			Annotation{X: 2.0, Y: 2.0, Label: "2.0"},
			Annotation{X: 3.0, Y: 3.0, Label: "3.0"},
			Annotation{X: 4.0, Y: 4.0, Label: "4.0"},
		},
	}

	r, err := PNG(110, 110)
	assert.Nil(err)

	f, err := GetDefaultFont()
	assert.Nil(err)

	xrange := Range{
		Min:    1.0,
		Max:    4.0,
		Domain: 100,
	}
	yrange := Range{
		Min:    1.0,
		Max:    4.0,
		Domain: 100,
	}

	cb := Box{
		Top:    5,
		Left:   5,
		Right:  105,
		Bottom: 105,
		Height: 100,
		Width:  100,
	}
	sd := Style{
		FontSize: 10.0,
		Font:     f,
	}

	as.Render(r, cb, xrange, yrange, sd)

	rr, isRaster := r.(*rasterRenderer)
	assert.True(isRaster)
	assert.NotNil(rr)

	c := rr.i.At(39, 70)
	converted, isRGBA := color.RGBAModel.Convert(c).(color.RGBA)
	assert.True(isRGBA)
	assert.Equal(3, converted.R)
	assert.Equal(3, converted.G)
	assert.Equal(3, converted.B)
}