package/gonum
1
0
Fork 0
mirror of https://github.com/gonum/gonum.git synced 2025-10-07 08:01:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

stat/combin: allow user allocation for Cartesian product

Browse Source
This commit is contained in:
Dan Kortschak
2019-10-24 19:27:42 +10:30
parent 7e1144f373
commit f54cad9e6d
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
stat/combin/combin.go
View File

@@ -341,8 +341,8 @@ func (g *CartesianGenerator) Next() bool {
// Product generates one product of the cartesian set according to the current index which is increased by Next(). // Product generates one product of the cartesian set according to the current index which is increased by Next().
// Next needs to be called at least one time before this method, otherwise it will panic. // Next needs to be called at least one time before this method, otherwise it will panic.
func (g *CartesianGenerator) Product() []int { func (g *CartesianGenerator) Product(dst []int) []int {
return SubFor(nil, g.idx, g.lens) return SubFor(dst, g.idx, g.lens)
} }
// IdxFor converts a multi-dimensional index into a linear index for a // IdxFor converts a multi-dimensional index into a linear index for a

2
stat/combin/combin_test.go
View File

@@ -318,7 +318,7 @@ func TestCartesianGenerator(t *testing.T) {
gen := NewCartesianGenerator([]int{1, 2, 3}) gen := NewCartesianGenerator([]int{1, 2, 3})
iterations := 0 iterations := 0
for gen.Next() { for gen.Next() {
got := gen.Product() got := gen.Product(nil)
if !reflect.DeepEqual(got, want[iterations]) { if !reflect.DeepEqual(got, want[iterations]) {
t.Errorf("Cartesian product does not match. want: %v got: %v", want[iterations], got) t.Errorf("Cartesian product does not match. want: %v got: %v", want[iterations], got)
} }

2
stat/combin/combinations_example_test.go
View File

@@ -39,7 +39,7 @@ func ExampleCartesianGenerator() {
// Now loop over all products. // Now loop over all products.
var i int var i int
for gen.Next() { for gen.Next() {
fmt.Println(i, gen.Product()) fmt.Println(i, gen.Product(nil))
i++ i++
} }