-
Notifications
You must be signed in to change notification settings - Fork 0
/
sheety_test.go
78 lines (73 loc) · 1.33 KB
/
sheety_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package sheety
import (
"testing"
"fmt"
)
type Student struct {
ID float64
Grade float64
}
var studentsExpected = []Student{
{
ID: 43875,
Grade: 84,
},
{
ID: 20347,
Grade: 72,
},
{
ID: 83274,
Grade: 83,
},
{
ID: 72345,
Grade: 99,
},
}
func TestXLSX(t *testing.T) {
s, err := OpenSpreadsheet("grades.xlsx")
if err != nil {
t.Fatal(err)
}
students := make([]*Student, 0, 3)
s.ReadRows(3, &students, map[int]string{
1: "ID",
2: "Grade",
})
for i, expected := range studentsExpected {
got := students[i]
if expected.ID != got.ID {
t.Fatalf("length: exp=%v, got=%v", expected.ID, got.ID)
}
if expected.Grade != got.Grade {
t.Fatal(fmt.Sprintf("exp=%v, got=%v", expected.Grade, got.Grade))
}
}
}
func TestCSV(t *testing.T) {
s, err := OpenCSV("grades.csv")
if err != nil {
t.Fatal(err)
}
students := make([]*Student, 0, 3)
err = s.ReadRows(1, &students, map[int]string{
1: "ID",
2: "Grade",
})
if err != nil {
t.Fatal(err)
}
if len(students) > 4 {
t.Fatalf("exp=%v got=%v", 4, len(students))
}
for i, expected := range studentsExpected {
got := students[i]
if expected.ID != got.ID {
t.Fatalf("length: exp=%v, got=%v", expected.ID, got.ID)
}
if expected.Grade != got.Grade {
t.Fatal(fmt.Sprintf("exp=%v, got=%v", expected.Grade, got.Grade))
}
}
}