-
Notifications
You must be signed in to change notification settings - Fork 0
/
value_sort_test.go
102 lines (93 loc) · 2.55 KB
/
value_sort_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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package dataframe
import (
"fmt"
"sort"
"testing"
"time"
)
func TestByStringAscending(t *testing.T) {
vs := []Value{}
for i := 0; i < 100; i++ {
vs = append(vs, NewStringValue(fmt.Sprintf("%d", i)))
}
sort.Sort(ByStringAscending(vs))
if !vs[0].EqualTo(NewStringValue("0")) {
t.Fatalf("expected '0', got %v", vs[0])
}
}
func TestByStringDescending(t *testing.T) {
vs := []Value{}
for i := 0; i < 100; i++ {
vs = append(vs, NewStringValue(fmt.Sprintf("%d", i)))
}
sort.Sort(ByStringDescending(vs))
if !vs[0].EqualTo(NewStringValue("99")) {
t.Fatalf("expected '99', got %v", vs[0])
}
}
func TestByStringDescendingTime(t *testing.T) {
now := time.Now()
vs := []Value{}
for i := 0; i < 100; i++ {
t := now.Add(time.Duration(i) * time.Second).String()
vs = append(vs, NewStringValue(t))
}
sort.Sort(ByStringDescending(vs))
if !vs[0].EqualTo(NewStringValue(now.Add(time.Duration(99) * time.Second).String())) {
t.Fatalf("expected '99', got %v", vs[0])
}
}
func TestByStringDescendingTimeColumn(t *testing.T) {
now := time.Now()
vs := []Value{}
for i := 0; i < 100; i++ {
t := now.Add(time.Duration(i) * time.Second)
vs = append(vs, NewTimeValue(t))
}
sort.Sort(ByStringDescending(vs))
if !vs[0].EqualTo(NewTimeValue(now.Add(time.Duration(99) * time.Second))) {
t.Fatalf("expected '99', got %v", vs[0])
}
}
func TestByFloat64Ascending(t *testing.T) {
vs := []Value{}
for i := 0; i < 100; i++ {
vs = append(vs, NewStringValue(fmt.Sprintf("%d", i)))
}
sort.Sort(ByFloat64Ascending(vs))
if !vs[0].EqualTo(NewStringValue("0")) {
t.Fatalf("expected '0', got %v", vs[0])
}
}
func TestByFloat64Descending(t *testing.T) {
vs := []Value{}
for i := 0; i < 100; i++ {
vs = append(vs, NewStringValue(fmt.Sprintf("%d", i)))
}
vs = append(vs, NewStringValue("199.9"))
sort.Sort(ByFloat64Descending(vs))
if !vs[0].EqualTo(NewStringValue("199.9")) {
t.Fatalf("expected '199.9', got %v", vs[0])
}
}
func TestByDurationAscending(t *testing.T) {
vs := []Value{}
for i := 0; i < 100; i++ {
vs = append(vs, NewStringValue(fmt.Sprintf("%s", time.Duration(i)*time.Second)))
}
sort.Sort(ByDurationAscending(vs))
if !vs[0].EqualTo(NewStringValue("0s")) {
t.Fatalf("expected '0s', got %v", vs[0])
}
}
func TestByDurationDescending(t *testing.T) {
vs := []Value{}
for i := 0; i < 100; i++ {
vs = append(vs, NewStringValue(fmt.Sprintf("%s", time.Duration(i)*time.Second)))
}
vs = append(vs, NewStringValue("200h"))
sort.Sort(ByDurationDescending(vs))
if !vs[0].EqualTo(NewStringValue("200h")) {
t.Fatalf("expected '200h', got %v", vs[0])
}
}