summaryrefslogtreecommitdiff
path: root/sparkline_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'sparkline_test.go')
-rw-r--r--sparkline_test.go310
1 files changed, 310 insertions, 0 deletions
diff --git a/sparkline_test.go b/sparkline_test.go
new file mode 100644
index 0000000..4b059ab
--- /dev/null
+++ b/sparkline_test.go
@@ -0,0 +1,310 @@
+package excelize
+
+import (
+ "fmt"
+ "path/filepath"
+ "testing"
+
+ "github.com/stretchr/testify/assert"
+)
+
+func TestAddSparkline(t *testing.T) {
+ f := prepareSparklineDataset()
+
+ // Set the columns widths to make the output clearer
+ style, err := f.NewStyle(`{"font":{"bold":true}}`)
+ assert.NoError(t, err)
+ assert.NoError(t, f.SetCellStyle("Sheet1", "A1", "B1", style))
+ assert.NoError(t, f.SetSheetViewOptions("Sheet1", 0, ZoomScale(150)))
+
+ assert.NoError(t, f.SetColWidth("Sheet1", "A", "A", 14))
+ assert.NoError(t, f.SetColWidth("Sheet1", "B", "B", 50))
+ // Headings
+ assert.NoError(t, f.SetCellValue("Sheet1", "A1", "Sparkline"))
+ assert.NoError(t, f.SetCellValue("Sheet1", "B1", "Description"))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B2", `A default "line" sparkline.`))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A2"},
+ Range: []string{"Sheet3!A1:J1"},
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B3", `A default "column" sparkline.`))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A3"},
+ Range: []string{"Sheet3!A2:J2"},
+ Type: "column",
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B4", `A default "win/loss" sparkline.`))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A4"},
+ Range: []string{"Sheet3!A3:J3"},
+ Type: "win_loss",
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B6", "Line with markers."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A6"},
+ Range: []string{"Sheet3!A1:J1"},
+ Markers: true,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B7", "Line with high and low points."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A7"},
+ Range: []string{"Sheet3!A1:J1"},
+ High: true,
+ Low: true,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B8", "Line with first and last point markers."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A8"},
+ Range: []string{"Sheet3!A1:J1"},
+ First: true,
+ Last: true,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B9", "Line with negative point markers."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A9"},
+ Range: []string{"Sheet3!A1:J1"},
+ Negative: true,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B10", "Line with axis."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A10"},
+ Range: []string{"Sheet3!A1:J1"},
+ Axis: true,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B12", "Column with default style (1)."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A12"},
+ Range: []string{"Sheet3!A2:J2"},
+ Type: "column",
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B13", "Column with style 2."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A13"},
+ Range: []string{"Sheet3!A2:J2"},
+ Type: "column",
+ Style: 2,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B14", "Column with style 3."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A14"},
+ Range: []string{"Sheet3!A2:J2"},
+ Type: "column",
+ Style: 3,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B15", "Column with style 4."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A15"},
+ Range: []string{"Sheet3!A2:J2"},
+ Type: "column",
+ Style: 4,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B16", "Column with style 5."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A16"},
+ Range: []string{"Sheet3!A2:J2"},
+ Type: "column",
+ Style: 5,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B17", "Column with style 6."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A17"},
+ Range: []string{"Sheet3!A2:J2"},
+ Type: "column",
+ Style: 6,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B18", "Column with a user defined color."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A18"},
+ Range: []string{"Sheet3!A2:J2"},
+ Type: "column",
+ SeriesColor: "#E965E0",
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B20", "A win/loss sparkline."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A20"},
+ Range: []string{"Sheet3!A3:J3"},
+ Type: "win_loss",
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B21", "A win/loss sparkline with negative points highlighted."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A21"},
+ Range: []string{"Sheet3!A3:J3"},
+ Type: "win_loss",
+ Negative: true,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B23", "A left to right column (the default)."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A23"},
+ Range: []string{"Sheet3!A4:J4"},
+ Type: "column",
+ Style: 20,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B24", "A right to left column."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A24"},
+ Range: []string{"Sheet3!A4:J4"},
+ Type: "column",
+ Style: 20,
+ Reverse: true,
+ }))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B25", "Sparkline and text in one cell."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A25"},
+ Range: []string{"Sheet3!A4:J4"},
+ Type: "column",
+ Style: 20,
+ }))
+ assert.NoError(t, f.SetCellValue("Sheet1", "A25", "Growth"))
+
+ assert.NoError(t, f.SetCellValue("Sheet1", "B27", "A grouped sparkline. Changes are applied to all three."))
+ assert.NoError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A27", "A28", "A29"},
+ Range: []string{"Sheet3!A5:J5", "Sheet3!A6:J6", "Sheet3!A7:J7"},
+ Markers: true,
+ }))
+
+ // Sheet2 sections
+ assert.NoError(t, f.AddSparkline("Sheet2", &SparklineOption{
+ Location: []string{"F3"},
+ Range: []string{"Sheet2!A3:E3"},
+ Type: "win_loss",
+ Negative: true,
+ }))
+
+ assert.NoError(t, f.AddSparkline("Sheet2", &SparklineOption{
+ Location: []string{"F1"},
+ Range: []string{"Sheet2!A1:E1"},
+ Markers: true,
+ }))
+
+ assert.NoError(t, f.AddSparkline("Sheet2", &SparklineOption{
+ Location: []string{"F2"},
+ Range: []string{"Sheet2!A2:E2"},
+ Type: "column",
+ Style: 12,
+ }))
+
+ assert.NoError(t, f.AddSparkline("Sheet2", &SparklineOption{
+ Location: []string{"F3"},
+ Range: []string{"Sheet2!A3:E3"},
+ Type: "win_loss",
+ Negative: true,
+ }))
+
+ // Save xlsx file by the given path.
+ assert.NoError(t, f.SaveAs(filepath.Join("test", "TestAddSparkline.xlsx")))
+
+ // Test error exceptions
+ assert.EqualError(t, f.AddSparkline("SheetN", &SparklineOption{
+ Location: []string{"F3"},
+ Range: []string{"Sheet2!A3:E3"},
+ }), "sheet SheetN is not exist")
+
+ assert.EqualError(t, f.AddSparkline("Sheet1", nil), "parameter is required")
+
+ assert.EqualError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Range: []string{"Sheet2!A3:E3"},
+ }), `parameter 'Location' is required`)
+
+ assert.EqualError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"F3"},
+ }), `parameter 'Range' is required`)
+
+ assert.EqualError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"F2", "F3"},
+ Range: []string{"Sheet2!A3:E3"},
+ }), `must have the same number of 'Location' and 'Range' parameters`)
+
+ assert.EqualError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"F3"},
+ Range: []string{"Sheet2!A3:E3"},
+ Type: "unknown_type",
+ }), `parameter 'Type' must be 'line', 'column' or 'win_loss'`)
+
+ assert.EqualError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"F3"},
+ Range: []string{"Sheet2!A3:E3"},
+ Style: -1,
+ }), `parameter 'Style' must betweent 0-35`)
+
+ assert.EqualError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"F3"},
+ Range: []string{"Sheet2!A3:E3"},
+ Style: -1,
+ }), `parameter 'Style' must betweent 0-35`)
+
+ f.Sheet["xl/worksheets/sheet1.xml"].ExtLst.Ext = `<extLst>
+ <ext x14="http://schemas.microsoft.com/office/spreadsheetml/2009/9/main" uri="{05C60535-1F16-4fd2-B633-F4F36F0B64E0}">
+ <x14:sparklineGroups
+ xmlns:xm="http://schemas.microsoft.com/office/excel/2006/main">
+ <x14:sparklineGroup>
+ </x14:sparklines>
+ </x14:sparklineGroup>
+ </x14:sparklineGroups>
+ </ext>
+ </extLst>`
+ assert.EqualError(t, f.AddSparkline("Sheet1", &SparklineOption{
+ Location: []string{"A2"},
+ Range: []string{"Sheet3!A1:J1"},
+ }), "XML syntax error on line 6: element <sparklineGroup> closed by </sparklines>")
+}
+
+func TestAppendSparkline(t *testing.T) {
+ // Test unsupport charset.
+ f := NewFile()
+ ws, err := f.workSheetReader("Sheet1")
+ assert.NoError(t, err)
+ ws.ExtLst = &xlsxExtLst{Ext: string(MacintoshCyrillicCharset)}
+ assert.EqualError(t, f.appendSparkline(ws, &xlsxX14SparklineGroup{}, &xlsxX14SparklineGroups{}), "XML syntax error on line 1: invalid UTF-8")
+}
+
+func prepareSparklineDataset() *File {
+ f := NewFile()
+ sheet2 := [][]int{
+ {-2, 2, 3, -1, 0},
+ {30, 20, 33, 20, 15},
+ {1, -1, -1, 1, -1},
+ }
+ sheet3 := [][]int{
+ {-2, 2, 3, -1, 0, -2, 3, 2, 1, 0},
+ {30, 20, 33, 20, 15, 5, 5, 15, 10, 15},
+ {1, 1, -1, -1, 1, -1, 1, 1, 1, -1},
+ {5, 6, 7, 10, 15, 20, 30, 50, 70, 100},
+ {-2, 2, 3, -1, 0, -2, 3, 2, 1, 0},
+ {3, -1, 0, -2, 3, 2, 1, 0, 2, 1},
+ {0, -2, 3, 2, 1, 0, 1, 2, 3, 1},
+ }
+ f.NewSheet("Sheet2")
+ f.NewSheet("Sheet3")
+ for row, data := range sheet2 {
+ if err := f.SetSheetRow("Sheet2", fmt.Sprintf("A%d", row+1), &data); err != nil {
+ fmt.Println(err)
+ }
+ }
+ for row, data := range sheet3 {
+ if err := f.SetSheetRow("Sheet3", fmt.Sprintf("A%d", row+1), &data); err != nil {
+ fmt.Println(err)
+ }
+ }
+ return f
+}
generated by cgit v1.2.1 (git 2.18.0) at 2025-04-19 12:47:08 +0000