mirror of
https://github.com/gogf/gf.git
synced 2025-04-05 03:05:05 +08:00
867 lines
21 KiB
Go
867 lines
21 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
||
//
|
||
// This Source Code Form is subject to the terms of the MIT License.
|
||
// If a copy of the MIT was not distributed with this file,
|
||
// You can obtain one at https://github.com/gogf/gf.
|
||
|
||
// go test *.go
|
||
|
||
package garray_test
|
||
|
||
import (
|
||
"testing"
|
||
"time"
|
||
|
||
"github.com/gogf/gf/v2/container/garray"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/internal/empty"
|
||
"github.com/gogf/gf/v2/internal/json"
|
||
"github.com/gogf/gf/v2/test/gtest"
|
||
"github.com/gogf/gf/v2/util/gconv"
|
||
)
|
||
|
||
func Test_IntArray_Basic(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
expect := []int{0, 1, 2, 3}
|
||
expect2 := []int{}
|
||
array := garray.NewIntArrayFrom(expect)
|
||
array2 := garray.NewIntArrayFrom(expect2)
|
||
t.Assert(array.Slice(), expect)
|
||
t.Assert(array.Interfaces(), expect)
|
||
array.Set(0, 100)
|
||
|
||
v, ok := array.Get(0)
|
||
t.Assert(v, 100)
|
||
t.Assert(ok, true)
|
||
|
||
v, ok = array.Get(1)
|
||
t.Assert(v, 1)
|
||
t.Assert(ok, true)
|
||
|
||
t.Assert(array.Search(100), 0)
|
||
t.Assert(array2.Search(100), -1)
|
||
t.Assert(array.Contains(100), true)
|
||
|
||
v, ok = array.Remove(0)
|
||
t.Assert(v, 100)
|
||
t.Assert(ok, true)
|
||
|
||
v, ok = array.Remove(-1)
|
||
t.Assert(v, 0)
|
||
t.Assert(ok, false)
|
||
|
||
v, ok = array.Remove(100000)
|
||
t.Assert(v, 0)
|
||
t.Assert(ok, false)
|
||
|
||
t.Assert(array.Contains(100), false)
|
||
array.Append(4)
|
||
t.Assert(array.Len(), 4)
|
||
array.InsertBefore(0, 100)
|
||
array.InsertAfter(0, 200)
|
||
t.Assert(array.Slice(), []int{100, 200, 1, 2, 3, 4})
|
||
array.InsertBefore(5, 300)
|
||
array.InsertAfter(6, 400)
|
||
t.Assert(array.Slice(), []int{100, 200, 1, 2, 3, 300, 4, 400})
|
||
t.Assert(array.Clear().Len(), 0)
|
||
err := array.InsertBefore(99, 300)
|
||
t.AssertNE(err, nil)
|
||
err = array.InsertAfter(99, 400)
|
||
t.AssertNE(err, nil)
|
||
})
|
||
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayFrom([]int{0, 1, 2, 3})
|
||
copyArray := array.DeepCopy().(*garray.IntArray)
|
||
copyArray.Set(0, 1)
|
||
cval, _ := copyArray.Get(0)
|
||
val, _ := array.Get(0)
|
||
t.AssertNE(cval, val)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Sort(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
expect1 := []int{0, 1, 2, 3}
|
||
expect2 := []int{3, 2, 1, 0}
|
||
array := garray.NewIntArray()
|
||
array2 := garray.NewIntArray(true)
|
||
for i := 3; i >= 0; i-- {
|
||
array.Append(i)
|
||
array2.Append(i)
|
||
}
|
||
array.Sort()
|
||
t.Assert(array.Slice(), expect1)
|
||
array.Sort(true)
|
||
t.Assert(array.Slice(), expect2)
|
||
t.Assert(array2.Slice(), expect2)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Unique(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
expect := []int{1, 2, 3, 4, 5, 3, 2, 2, 3, 5, 5}
|
||
array := garray.NewIntArrayFrom(expect)
|
||
t.Assert(array.Unique().Slice(), []int{1, 2, 3, 4, 5})
|
||
array2 := garray.NewIntArrayFrom([]int{})
|
||
t.Assert(array2.Unique().Slice(), []int{})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_PushAndPop(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
expect := []int{0, 1, 2, 3}
|
||
array := garray.NewIntArrayFrom(expect)
|
||
t.Assert(array.Slice(), expect)
|
||
|
||
v, ok := array.PopLeft()
|
||
t.Assert(v, 0)
|
||
t.Assert(ok, true)
|
||
|
||
v, ok = array.PopRight()
|
||
t.Assert(v, 3)
|
||
t.Assert(ok, true)
|
||
|
||
v, ok = array.PopRand()
|
||
t.AssertIN(v, []int{1, 2})
|
||
t.Assert(ok, true)
|
||
|
||
v, ok = array.PopRand()
|
||
t.AssertIN(v, []int{1, 2})
|
||
t.Assert(ok, true)
|
||
|
||
v, ok = array.PopRand()
|
||
t.Assert(v, 0)
|
||
t.Assert(ok, false)
|
||
|
||
t.Assert(array.Len(), 0)
|
||
array.PushLeft(1).PushRight(2)
|
||
t.Assert(array.Slice(), []int{1, 2})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_PopLeftsAndPopRights(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArray()
|
||
|
||
v, ok := array.PopLeft()
|
||
t.Assert(v, 0)
|
||
t.Assert(ok, false)
|
||
|
||
t.Assert(array.PopLefts(10), nil)
|
||
|
||
v, ok = array.PopRight()
|
||
t.Assert(v, 0)
|
||
t.Assert(ok, false)
|
||
|
||
t.Assert(array.PopRights(10), nil)
|
||
|
||
v, ok = array.PopRand()
|
||
t.Assert(v, 0)
|
||
t.Assert(ok, false)
|
||
|
||
t.Assert(array.PopRands(10), nil)
|
||
})
|
||
|
||
gtest.C(t, func(t *gtest.T) {
|
||
value1 := []int{0, 1, 2, 3, 4, 5, 6}
|
||
value2 := []int{0, 1, 2, 3, 4, 5, 6}
|
||
array1 := garray.NewIntArrayFrom(value1)
|
||
array2 := garray.NewIntArrayFrom(value2)
|
||
t.Assert(array1.PopLefts(2), []int{0, 1})
|
||
t.Assert(array1.Slice(), []int{2, 3, 4, 5, 6})
|
||
t.Assert(array1.PopRights(2), []int{5, 6})
|
||
t.Assert(array1.Slice(), []int{2, 3, 4})
|
||
t.Assert(array1.PopRights(20), []int{2, 3, 4})
|
||
t.Assert(array1.Slice(), []int{})
|
||
t.Assert(array2.PopLefts(20), []int{0, 1, 2, 3, 4, 5, 6})
|
||
t.Assert(array2.Slice(), []int{})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Range(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
value1 := []int{0, 1, 2, 3, 4, 5, 6}
|
||
array1 := garray.NewIntArrayFrom(value1)
|
||
array2 := garray.NewIntArrayFrom(value1, true)
|
||
t.Assert(array1.Range(0, 1), []int{0})
|
||
t.Assert(array1.Range(1, 2), []int{1})
|
||
t.Assert(array1.Range(0, 2), []int{0, 1})
|
||
t.Assert(array1.Range(10, 2), nil)
|
||
t.Assert(array1.Range(-1, 10), value1)
|
||
t.Assert(array2.Range(1, 2), []int{1})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Merge(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
func1 := func(v1, v2 interface{}) int {
|
||
if gconv.Int(v1) < gconv.Int(v2) {
|
||
return 0
|
||
}
|
||
return 1
|
||
}
|
||
|
||
n1 := []int{0, 1, 2, 3}
|
||
n2 := []int{4, 5, 6, 7}
|
||
i1 := []interface{}{"1", "2"}
|
||
s1 := []string{"a", "b", "c"}
|
||
s2 := []string{"e", "f"}
|
||
a1 := garray.NewIntArrayFrom(n1)
|
||
a2 := garray.NewIntArrayFrom(n2)
|
||
a3 := garray.NewArrayFrom(i1)
|
||
a4 := garray.NewStrArrayFrom(s1)
|
||
|
||
a5 := garray.NewSortedStrArrayFrom(s2)
|
||
a6 := garray.NewSortedIntArrayFrom([]int{1, 2, 3})
|
||
|
||
a7 := garray.NewSortedStrArrayFrom(s1)
|
||
a8 := garray.NewSortedArrayFrom([]interface{}{4, 5}, func1)
|
||
|
||
t.Assert(a1.Merge(a2).Slice(), []int{0, 1, 2, 3, 4, 5, 6, 7})
|
||
t.Assert(a1.Merge(a3).Len(), 10)
|
||
t.Assert(a1.Merge(a4).Len(), 13)
|
||
t.Assert(a1.Merge(a5).Len(), 15)
|
||
t.Assert(a1.Merge(a6).Len(), 18)
|
||
t.Assert(a1.Merge(a7).Len(), 21)
|
||
t.Assert(a1.Merge(a8).Len(), 23)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Fill(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{0}
|
||
a2 := []int{0}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
array2 := garray.NewIntArrayFrom(a2)
|
||
t.Assert(array1.Fill(1, 2, 100), nil)
|
||
t.Assert(array1.Slice(), []int{0, 100, 100})
|
||
|
||
t.Assert(array2.Fill(0, 2, 100), nil)
|
||
t.Assert(array2.Slice(), []int{100, 100})
|
||
|
||
t.AssertNE(array2.Fill(-1, 2, 100), nil)
|
||
t.Assert(array2.Slice(), []int{100, 100})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_PopLeft(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayFrom(g.SliceInt{1, 2, 3})
|
||
v, ok := array.PopLeft()
|
||
t.Assert(v, 1)
|
||
t.Assert(ok, true)
|
||
t.Assert(array.Len(), 2)
|
||
v, ok = array.PopLeft()
|
||
t.Assert(v, 2)
|
||
t.Assert(ok, true)
|
||
t.Assert(array.Len(), 1)
|
||
v, ok = array.PopLeft()
|
||
t.Assert(v, 3)
|
||
t.Assert(ok, true)
|
||
t.Assert(array.Len(), 0)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_PopRight(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayFrom(g.SliceInt{1, 2, 3})
|
||
|
||
v, ok := array.PopRight()
|
||
t.Assert(v, 3)
|
||
t.Assert(ok, true)
|
||
t.Assert(array.Len(), 2)
|
||
|
||
v, ok = array.PopRight()
|
||
t.Assert(v, 2)
|
||
t.Assert(ok, true)
|
||
t.Assert(array.Len(), 1)
|
||
|
||
v, ok = array.PopRight()
|
||
t.Assert(v, 1)
|
||
t.Assert(ok, true)
|
||
t.Assert(array.Len(), 0)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_PopLefts(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayFrom(g.SliceInt{1, 2, 3})
|
||
t.Assert(array.PopLefts(2), g.Slice{1, 2})
|
||
t.Assert(array.Len(), 1)
|
||
t.Assert(array.PopLefts(2), g.Slice{3})
|
||
t.Assert(array.Len(), 0)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_PopRights(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayFrom(g.SliceInt{1, 2, 3})
|
||
t.Assert(array.PopRights(2), g.Slice{2, 3})
|
||
t.Assert(array.Len(), 1)
|
||
t.Assert(array.PopLefts(2), g.Slice{1})
|
||
t.Assert(array.Len(), 0)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Chunk(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 4, 5}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
chunks := array1.Chunk(2)
|
||
t.Assert(len(chunks), 3)
|
||
t.Assert(chunks[0], []int{1, 2})
|
||
t.Assert(chunks[1], []int{3, 4})
|
||
t.Assert(chunks[2], []int{5})
|
||
t.Assert(array1.Chunk(0), nil)
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 4, 5}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
chunks := array1.Chunk(3)
|
||
t.Assert(len(chunks), 2)
|
||
t.Assert(chunks[0], []int{1, 2, 3})
|
||
t.Assert(chunks[1], []int{4, 5})
|
||
t.Assert(array1.Chunk(0), nil)
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 4, 5, 6}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
chunks := array1.Chunk(2)
|
||
t.Assert(len(chunks), 3)
|
||
t.Assert(chunks[0], []int{1, 2})
|
||
t.Assert(chunks[1], []int{3, 4})
|
||
t.Assert(chunks[2], []int{5, 6})
|
||
t.Assert(array1.Chunk(0), nil)
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 4, 5, 6}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
chunks := array1.Chunk(3)
|
||
t.Assert(len(chunks), 2)
|
||
t.Assert(chunks[0], []int{1, 2, 3})
|
||
t.Assert(chunks[1], []int{4, 5, 6})
|
||
t.Assert(array1.Chunk(0), nil)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Pad(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{0}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
t.Assert(array1.Pad(3, 1).Slice(), []int{0, 1, 1})
|
||
t.Assert(array1.Pad(-4, 1).Slice(), []int{1, 0, 1, 1})
|
||
t.Assert(array1.Pad(3, 1).Slice(), []int{1, 0, 1, 1})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_SubSlice(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{0, 1, 2, 3, 4, 5, 6}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
array2 := garray.NewIntArrayFrom(a1, true)
|
||
t.Assert(array1.SubSlice(6), []int{6})
|
||
t.Assert(array1.SubSlice(5), []int{5, 6})
|
||
t.Assert(array1.SubSlice(8), nil)
|
||
t.Assert(array1.SubSlice(0, 2), []int{0, 1})
|
||
t.Assert(array1.SubSlice(2, 2), []int{2, 3})
|
||
t.Assert(array1.SubSlice(5, 8), []int{5, 6})
|
||
t.Assert(array1.SubSlice(-1, 1), []int{6})
|
||
t.Assert(array1.SubSlice(-1, 9), []int{6})
|
||
t.Assert(array1.SubSlice(-2, 3), []int{5, 6})
|
||
t.Assert(array1.SubSlice(-7, 3), []int{0, 1, 2})
|
||
t.Assert(array1.SubSlice(-8, 3), nil)
|
||
t.Assert(array1.SubSlice(-1, -3), []int{3, 4, 5})
|
||
t.Assert(array1.SubSlice(-9, 3), nil)
|
||
t.Assert(array1.SubSlice(1, -1), []int{0})
|
||
t.Assert(array1.SubSlice(1, -3), nil)
|
||
t.Assert(array2.SubSlice(0, 2), []int{0, 1})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Rand(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{0, 1, 2, 3, 4, 5, 6}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
t.Assert(len(array1.Rands(2)), 2)
|
||
t.Assert(len(array1.Rands(10)), 10)
|
||
t.AssertIN(array1.Rands(1)[0], a1)
|
||
|
||
v, ok := array1.Rand()
|
||
t.AssertIN(v, a1)
|
||
t.Assert(ok, true)
|
||
|
||
array2 := garray.NewIntArrayFrom([]int{})
|
||
v, ok = array2.Rand()
|
||
t.Assert(v, 0)
|
||
t.Assert(ok, false)
|
||
|
||
intSlices := array2.Rands(1)
|
||
t.Assert(intSlices, nil)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_PopRands(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{100, 200, 300, 400, 500, 600}
|
||
array := garray.NewIntArrayFrom(a1)
|
||
ns1 := array.PopRands(2)
|
||
t.AssertIN(ns1, []int{100, 200, 300, 400, 500, 600})
|
||
t.Assert(len(ns1), 2)
|
||
|
||
ns2 := array.PopRands(7)
|
||
t.Assert(len(ns2), 4)
|
||
t.AssertIN(ns2, []int{100, 200, 300, 400, 500, 600})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Shuffle(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{0, 1, 2, 3, 4, 5, 6}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
t.Assert(array1.Shuffle().Len(), 7)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Reverse(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{0, 1, 2, 3, 4, 5, 6}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
t.Assert(array1.Reverse().Slice(), []int{6, 5, 4, 3, 2, 1, 0})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Join(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{0, 1, 2, 3, 4, 5, 6}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
t.Assert(array1.Join("."), "0.1.2.3.4.5.6")
|
||
})
|
||
}
|
||
|
||
func TestIntArray_String(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{0, 1, 2, 3, 4, 5, 6}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
t.Assert(array1.String(), "[0,1,2,3,4,5,6]")
|
||
array1 = nil
|
||
t.Assert(array1.String(), "")
|
||
})
|
||
}
|
||
|
||
func TestIntArray_SetArray(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 5}
|
||
a2 := []int{6, 7}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
array1.SetArray(a2)
|
||
t.Assert(array1.Len(), 2)
|
||
t.Assert(array1, []int{6, 7})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Replace(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 5}
|
||
a2 := []int{6, 7}
|
||
a3 := []int{9, 10, 11, 12, 13}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
array1.Replace(a2)
|
||
t.Assert(array1, []int{6, 7, 3, 5})
|
||
|
||
array1.Replace(a3)
|
||
t.Assert(array1, []int{9, 10, 11, 12})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Clear(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 5}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
array1.Clear()
|
||
t.Assert(array1.Len(), 0)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Clone(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 5}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
array2 := array1.Clone()
|
||
t.Assert(array1, array2)
|
||
})
|
||
}
|
||
|
||
func TestArray_Get(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 5}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
v, ok := array1.Get(2)
|
||
t.Assert(v, 3)
|
||
t.Assert(ok, true)
|
||
t.Assert(array1.Len(), 4)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Sum(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 5}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
t.Assert(array1.Sum(), 11)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_CountValues(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 5, 3}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
m1 := array1.CountValues()
|
||
t.Assert(len(m1), 4)
|
||
t.Assert(m1[1], 1)
|
||
t.Assert(m1[3], 2)
|
||
})
|
||
}
|
||
|
||
func TestNewIntArrayFromCopy(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 5, 3}
|
||
array1 := garray.NewIntArrayFromCopy(a1)
|
||
t.Assert(array1.Len(), 5)
|
||
t.Assert(array1, a1)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Remove(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
a1 := []int{1, 2, 3, 5, 4}
|
||
array1 := garray.NewIntArrayFrom(a1)
|
||
v, ok := array1.Remove(1)
|
||
t.Assert(v, 2)
|
||
t.Assert(ok, true)
|
||
t.Assert(array1.Len(), 4)
|
||
|
||
v, ok = array1.Remove(0)
|
||
t.Assert(v, 1)
|
||
t.Assert(ok, true)
|
||
t.Assert(array1.Len(), 3)
|
||
|
||
v, ok = array1.Remove(2)
|
||
t.Assert(v, 4)
|
||
t.Assert(ok, true)
|
||
t.Assert(array1.Len(), 2)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_LockFunc(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
s1 := []int{1, 2, 3, 4}
|
||
a1 := garray.NewIntArrayFrom(s1, true)
|
||
|
||
ch1 := make(chan int64, 3)
|
||
ch2 := make(chan int64, 3)
|
||
// go1
|
||
go a1.LockFunc(func(n1 []int) { // 读写锁
|
||
time.Sleep(2 * time.Second) // 暂停2秒
|
||
n1[2] = 6
|
||
ch2 <- gconv.Int64(time.Now().UnixNano() / 1000 / 1000)
|
||
})
|
||
|
||
// go2
|
||
go func() {
|
||
time.Sleep(100 * time.Millisecond) // 故意暂停0.01秒,等go1执行锁后，再开始执行.
|
||
ch1 <- gconv.Int64(time.Now().UnixNano() / 1000 / 1000)
|
||
a1.Len()
|
||
ch1 <- gconv.Int64(time.Now().UnixNano() / 1000 / 1000)
|
||
}()
|
||
|
||
t1 := <-ch1
|
||
t2 := <-ch1
|
||
<-ch2 // 等待go1完成
|
||
|
||
// 防止ci抖动,以豪秒为单位
|
||
t.AssertGT(t2-t1, 20) // go1加的读写互斥锁，所go2读的时候被阻塞。
|
||
t.Assert(a1.Contains(6), true)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_SortFunc(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
s1 := []int{1, 4, 3, 2}
|
||
a1 := garray.NewIntArrayFrom(s1)
|
||
func1 := func(v1, v2 int) bool {
|
||
return v1 < v2
|
||
}
|
||
a11 := a1.SortFunc(func1)
|
||
t.Assert(a11, []int{1, 2, 3, 4})
|
||
|
||
})
|
||
}
|
||
|
||
func TestIntArray_RLockFunc(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
s1 := []int{1, 2, 3, 4}
|
||
a1 := garray.NewIntArrayFrom(s1, true)
|
||
|
||
ch1 := make(chan int64, 3)
|
||
ch2 := make(chan int64, 1)
|
||
// go1
|
||
go a1.RLockFunc(func(n1 []int) { // 读锁
|
||
time.Sleep(2 * time.Second) // 暂停1秒
|
||
n1[2] = 6
|
||
ch2 <- gconv.Int64(time.Now().UnixNano() / 1000 / 1000)
|
||
})
|
||
|
||
// go2
|
||
go func() {
|
||
time.Sleep(100 * time.Millisecond) // 故意暂停0.01秒,等go1执行锁后，再开始执行.
|
||
ch1 <- gconv.Int64(time.Now().UnixNano() / 1000 / 1000)
|
||
a1.Len()
|
||
ch1 <- gconv.Int64(time.Now().UnixNano() / 1000 / 1000)
|
||
}()
|
||
|
||
t1 := <-ch1
|
||
t2 := <-ch1
|
||
<-ch2 // 等待go1完成
|
||
|
||
// 防止ci抖动,以豪秒为单位
|
||
t.AssertLT(t2-t1, 20) // go1加的读锁，所go2读的时候，并没有阻塞。
|
||
t.Assert(a1.Contains(6), true)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Json(t *testing.T) {
|
||
// array pointer
|
||
gtest.C(t, func(t *gtest.T) {
|
||
s1 := []int{1, 4, 3, 2}
|
||
a1 := garray.NewIntArrayFrom(s1)
|
||
b1, err1 := json.Marshal(a1)
|
||
b2, err2 := json.Marshal(s1)
|
||
t.Assert(b1, b2)
|
||
t.Assert(err1, err2)
|
||
|
||
a2 := garray.NewIntArray()
|
||
err1 = json.UnmarshalUseNumber(b2, &a2)
|
||
t.AssertNil(err1)
|
||
t.Assert(a2.Slice(), s1)
|
||
|
||
var a3 garray.IntArray
|
||
err := json.UnmarshalUseNumber(b2, &a3)
|
||
t.AssertNil(err)
|
||
t.Assert(a3.Slice(), s1)
|
||
})
|
||
// array value
|
||
gtest.C(t, func(t *gtest.T) {
|
||
s1 := []int{1, 4, 3, 2}
|
||
a1 := *garray.NewIntArrayFrom(s1)
|
||
b1, err1 := json.Marshal(a1)
|
||
b2, err2 := json.Marshal(s1)
|
||
t.Assert(b1, b2)
|
||
t.Assert(err1, err2)
|
||
|
||
a2 := garray.NewIntArray()
|
||
err1 = json.UnmarshalUseNumber(b2, &a2)
|
||
t.AssertNil(err1)
|
||
t.Assert(a2.Slice(), s1)
|
||
|
||
var a3 garray.IntArray
|
||
err := json.UnmarshalUseNumber(b2, &a3)
|
||
t.AssertNil(err)
|
||
t.Assert(a3.Slice(), s1)
|
||
})
|
||
// array pointer
|
||
gtest.C(t, func(t *gtest.T) {
|
||
type User struct {
|
||
Name string
|
||
Scores *garray.IntArray
|
||
}
|
||
data := g.Map{
|
||
"Name": "john",
|
||
"Scores": []int{99, 100, 98},
|
||
}
|
||
b, err := json.Marshal(data)
|
||
t.AssertNil(err)
|
||
|
||
user := new(User)
|
||
err = json.UnmarshalUseNumber(b, user)
|
||
t.AssertNil(err)
|
||
t.Assert(user.Name, data["Name"])
|
||
t.Assert(user.Scores, data["Scores"])
|
||
})
|
||
// array value
|
||
gtest.C(t, func(t *gtest.T) {
|
||
type User struct {
|
||
Name string
|
||
Scores garray.IntArray
|
||
}
|
||
data := g.Map{
|
||
"Name": "john",
|
||
"Scores": []int{99, 100, 98},
|
||
}
|
||
b, err := json.Marshal(data)
|
||
t.AssertNil(err)
|
||
|
||
user := new(User)
|
||
err = json.UnmarshalUseNumber(b, user)
|
||
t.AssertNil(err)
|
||
t.Assert(user.Name, data["Name"])
|
||
t.Assert(user.Scores, data["Scores"])
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Iterator(t *testing.T) {
|
||
slice := g.SliceInt{10, 20, 30, 40}
|
||
array := garray.NewIntArrayFrom(slice)
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array.Iterator(func(k int, v int) bool {
|
||
t.Assert(v, slice[k])
|
||
return true
|
||
})
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array.IteratorAsc(func(k int, v int) bool {
|
||
t.Assert(v, slice[k])
|
||
return true
|
||
})
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array.IteratorDesc(func(k int, v int) bool {
|
||
t.Assert(v, slice[k])
|
||
return true
|
||
})
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
index := 0
|
||
array.Iterator(func(k int, v int) bool {
|
||
index++
|
||
return false
|
||
})
|
||
t.Assert(index, 1)
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
index := 0
|
||
array.IteratorAsc(func(k int, v int) bool {
|
||
index++
|
||
return false
|
||
})
|
||
t.Assert(index, 1)
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
index := 0
|
||
array.IteratorDesc(func(k int, v int) bool {
|
||
index++
|
||
return false
|
||
})
|
||
t.Assert(index, 1)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_RemoveValue(t *testing.T) {
|
||
slice := g.SliceInt{10, 20, 30, 40}
|
||
array := garray.NewIntArrayFrom(slice)
|
||
gtest.C(t, func(t *gtest.T) {
|
||
t.Assert(array.RemoveValue(99), false)
|
||
t.Assert(array.RemoveValue(20), true)
|
||
t.Assert(array.RemoveValue(10), true)
|
||
t.Assert(array.RemoveValue(20), false)
|
||
t.Assert(array.RemoveValue(88), false)
|
||
t.Assert(array.Len(), 2)
|
||
})
|
||
}
|
||
|
||
func TestIntArray_RemoveValues(t *testing.T) {
|
||
slice := g.SliceInt{10, 20, 30, 40}
|
||
array := garray.NewIntArrayFrom(slice)
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array.RemoveValues(10, 20, 40)
|
||
t.Assert(array.Slice(), g.SliceInt{30})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_UnmarshalValue(t *testing.T) {
|
||
type V struct {
|
||
Name string
|
||
Array *garray.IntArray
|
||
}
|
||
// JSON
|
||
gtest.C(t, func(t *gtest.T) {
|
||
var v *V
|
||
err := gconv.Struct(g.Map{
|
||
"name": "john",
|
||
"array": []byte(`[1,2,3]`),
|
||
}, &v)
|
||
t.AssertNil(err)
|
||
t.Assert(v.Name, "john")
|
||
t.Assert(v.Array.Slice(), g.Slice{1, 2, 3})
|
||
})
|
||
// Map
|
||
// gtest.C(t, func(t *gtest.T) {
|
||
// var v *V
|
||
// err := gconv.Struct(g.Map{
|
||
// "name": "john",
|
||
// "array": g.Slice{1, 2, 3},
|
||
// }, &v)
|
||
// t.AssertNil(err)
|
||
// t.Assert(v.Name, "john")
|
||
// t.Assert(v.Array.Slice(), g.Slice{1, 2, 3})
|
||
// })
|
||
}
|
||
|
||
func TestIntArray_Filter(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayFrom(g.SliceInt{0, 1, 2, 3, 4, 0})
|
||
t.Assert(array.Filter(func(index int, value int) bool {
|
||
return empty.IsEmpty(value)
|
||
}), g.SliceInt{1, 2, 3, 4})
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayFrom(g.SliceInt{1, 2, 3, 4})
|
||
t.Assert(array.Filter(func(index int, value int) bool {
|
||
return empty.IsEmpty(value)
|
||
|
||
}), g.SliceInt{1, 2, 3, 4})
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayFrom(g.SliceInt{1, 2, 3, 4})
|
||
t.Assert(array.Filter(func(index int, value int) bool {
|
||
return value%2 == 0
|
||
}), g.SliceInt{1, 3})
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayFrom(g.SliceInt{1, 2, 3, 4})
|
||
t.Assert(array.Filter(func(index int, value int) bool {
|
||
return value%2 == 1
|
||
}), g.SliceInt{2, 4})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_FilterEmpty(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayFrom(g.SliceInt{0, 1, 2, 3, 4, 0})
|
||
t.Assert(array.FilterEmpty(), g.SliceInt{1, 2, 3, 4})
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayFrom(g.SliceInt{1, 2, 3, 4})
|
||
t.Assert(array.FilterEmpty(), g.SliceInt{1, 2, 3, 4})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_Walk(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayFrom(g.SliceInt{1, 2})
|
||
t.Assert(array.Walk(func(value int) int {
|
||
return 10 + value
|
||
}), g.Slice{11, 12})
|
||
})
|
||
}
|
||
|
||
func TestIntArray_NewIntArrayRange(t *testing.T) {
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayRange(0, 128, 4)
|
||
t.Assert(array.String(), `[0,4,8,12,16,20,24,28,32,36,40,44,48,52,56,60,64,68,72,76,80,84,88,92,96,100,104,108,112,116,120,124,128]`)
|
||
})
|
||
gtest.C(t, func(t *gtest.T) {
|
||
array := garray.NewIntArrayRange(1, 128, 4)
|
||
t.Assert(array.String(), `[1,5,9,13,17,21,25,29,33,37,41,45,49,53,57,61,65,69,73,77,81,85,89,93,97,101,105,109,113,117,121,125]`)
|
||
})
|
||
}
|