github.com/zhiqiangxu/util@v0.0.0-20230112053021-0a7aee056cd5/math/partition_test.go (about)

     1  package math
     2  
     3  import (
     4  	"testing"
     5  
     6  	"gotest.tools/assert"
     7  )
     8  
     9  func TestPartition(t *testing.T) {
    10  	p1 := Partition(1)
    11  	p2 := Partition(2)
    12  	p3 := Partition(3)
    13  	p4 := Partition(4)
    14  	p5 := Partition(5)
    15  	p6 := Partition(6)
    16  
    17  	assert.Equal(t, p1.Uint64(), uint64(1))
    18  	assert.Equal(t, p2.Uint64(), uint64(2))
    19  	assert.Equal(t, p3.Uint64(), uint64(3))
    20  	assert.Equal(t, p4.Uint64(), uint64(5))
    21  	assert.Equal(t, p5.Uint64(), uint64(7))
    22  	assert.Equal(t, p6.Uint64(), uint64(11))
    23  }