mirror of
https://github.com/duke-git/lancet.git
synced 2026-02-04 12:52:28 +08:00
130 lines
2.6 KiB
Go
130 lines
2.6 KiB
Go
package datastructure
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/duke-git/lancet/v2/internal"
|
|
)
|
|
|
|
func TestSet_Add(t *testing.T) {
|
|
assert := internal.NewAssert(t, "TestSet_Add")
|
|
|
|
set := NewSet[int]()
|
|
set.Add(1, 2, 3)
|
|
|
|
expected := NewSet(1, 2, 3)
|
|
|
|
assert.Equal(true, set.Equal(expected))
|
|
}
|
|
|
|
func TestSet_Contain(t *testing.T) {
|
|
assert := internal.NewAssert(t, "TestSet_Contain")
|
|
|
|
set := NewSet[int]()
|
|
set.Add(1, 2, 3)
|
|
|
|
assert.Equal(true, set.Contain(1))
|
|
assert.Equal(false, set.Contain(4))
|
|
}
|
|
|
|
func TestSet_ContainAll(t *testing.T) {
|
|
assert := internal.NewAssert(t, "TestSet_ContainAll")
|
|
|
|
set1 := NewSet(1, 2, 3)
|
|
set2 := NewSet(1, 2)
|
|
set3 := NewSet(1, 2, 3, 4)
|
|
|
|
assert.Equal(true, set1.ContainAll(set2))
|
|
assert.Equal(false, set1.ContainAll(set3))
|
|
}
|
|
|
|
func TestSet_Clone(t *testing.T) {
|
|
assert := internal.NewAssert(t, "TestSet_Clone")
|
|
|
|
set1 := NewSet(1, 2, 3)
|
|
set2 := set1.Clone()
|
|
|
|
assert.Equal(true, set1.Size() == set2.Size())
|
|
assert.Equal(true, set1.ContainAll(set2))
|
|
}
|
|
|
|
func TestSet_Delete(t *testing.T) {
|
|
assert := internal.NewAssert(t, "TestSet_Delete")
|
|
|
|
set := NewSet[int]()
|
|
set.Add(1, 2, 3)
|
|
set.Delete(3)
|
|
|
|
expected := NewSet(1, 2)
|
|
|
|
assert.Equal(true, set.Equal(expected))
|
|
}
|
|
|
|
func TestSet_Equal(t *testing.T) {
|
|
assert := internal.NewAssert(t, "TestSet_Equal")
|
|
|
|
set1 := NewSet(1, 2, 3)
|
|
set2 := NewSet(1, 2, 3)
|
|
set3 := NewSet(1, 2, 3, 4)
|
|
|
|
assert.Equal(true, set1.Equal(set2))
|
|
assert.Equal(false, set1.Equal(set3))
|
|
}
|
|
|
|
func TestSet_Iterate(t *testing.T) {
|
|
assert := internal.NewAssert(t, "TestSet_Iterate")
|
|
|
|
set := NewSet(1, 2, 3)
|
|
arr := []int{}
|
|
set.Iterate(func(value int) {
|
|
arr = append(arr, value)
|
|
})
|
|
|
|
assert.Equal(3, len(arr))
|
|
}
|
|
|
|
func TestSet_IsEmpty(t *testing.T) {
|
|
assert := internal.NewAssert(t, "TestSet_IsEmpty")
|
|
|
|
set := NewSet[int]()
|
|
assert.Equal(true, set.IsEmpty())
|
|
}
|
|
|
|
func TestSet_Size(t *testing.T) {
|
|
assert := internal.NewAssert(t, "TestSet_Size")
|
|
|
|
set := NewSet(1, 2, 3)
|
|
assert.Equal(3, set.Size())
|
|
}
|
|
|
|
func TestSet_Values(t *testing.T) {
|
|
assert := internal.NewAssert(t, "TestSet_Values")
|
|
|
|
set := NewSet(1, 2, 3)
|
|
values := set.Values()
|
|
|
|
assert.Equal(3, len(values))
|
|
}
|
|
|
|
func TestSet_Union(t *testing.T) {
|
|
assert := internal.NewAssert(t, "TestSet_Union")
|
|
|
|
set1 := NewSet(1, 2, 3)
|
|
set2 := NewSet(2, 3, 4, 5)
|
|
expected := NewSet(1, 2, 3, 4, 5)
|
|
unionSet := set1.Union(set2)
|
|
|
|
assert.Equal(expected, unionSet)
|
|
}
|
|
|
|
func TestSet_Intersection(t *testing.T) {
|
|
assert := internal.NewAssert(t, "TestSet_Intersection")
|
|
|
|
set1 := NewSet(1, 2, 3)
|
|
set2 := NewSet(2, 3, 4, 5)
|
|
expected := NewSet(2, 3)
|
|
intersectionSet := set1.Intersection(set2)
|
|
|
|
assert.Equal(expected, intersectionSet)
|
|
}
|