49 lines
1 KiB
Go
49 lines
1 KiB
Go
|
package seq
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
assert "github.com/blendlabs/go-assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestRange(t *testing.T) {
|
||
|
|
assert := assert.New(t)
|
||
|
|
|
||
|
|
values := Range(1, 100)
|
||
|
|
assert.Len(values, 100)
|
||
|
|
assert.Equal(1, values[0])
|
||
|
|
assert.Equal(100, values[99])
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRangeWithStep(t *testing.T) {
|
||
|
|
assert := assert.New(t)
|
||
|
|
|
||
|
|
values := RangeWithStep(0, 100, 5)
|
||
|
|
assert.Equal(100, values[20])
|
||
|
|
assert.Len(values, 21)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRangeReversed(t *testing.T) {
|
||
|
|
assert := assert.New(t)
|
||
|
|
|
||
|
|
values := Range(10.0, 1.0)
|
||
|
|
assert.Equal(10, len(values))
|
||
|
|
assert.Equal(10.0, values[0])
|
||
|
|
assert.Equal(1.0, values[9])
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestValuesRegression(t *testing.T) {
|
||
|
|
assert := assert.New(t)
|
||
|
|
|
||
|
|
// note; this assumes a 1.0 step is implicitly set in the constructor.
|
||
|
|
linearProvider := NewLinear().WithStart(1.0).WithEnd(100.0)
|
||
|
|
assert.Equal(1, linearProvider.Start())
|
||
|
|
assert.Equal(100, linearProvider.End())
|
||
|
|
assert.Equal(100, linearProvider.Len())
|
||
|
|
|
||
|
|
values := Seq{Provider: linearProvider}.Array()
|
||
|
|
assert.Len(values, 100)
|
||
|
|
assert.Equal(1.0, values[0])
|
||
|
|
assert.Equal(100, values[99])
|
||
|
|
}
|