25 lines
614 B
Go
25 lines
614 B
Go
|
// Copyright (c) 2026 Petr Balvín <opensource@petrbalvin.org> (https://petrbalvin.org)
|
||
|
|
// SPDX-License-Identifier: MIT
|
||
|
|
|
||
|
|
package optim
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"sourcedock.dev/petrbalvin/tensor/internal/core"
|
||
|
|
)
|
||
|
|
|
||
|
|
// mustFloats builds a float array, failing the test on a bad shape.
|
||
|
|
// Without an explicit shape it defaults to a vector of len(vals).
|
||
|
|
func mustFloats(t *testing.T, vals []float64, shape ...int) *core.Array {
|
||
|
|
t.Helper()
|
||
|
|
if len(shape) == 0 {
|
||
|
|
shape = []int{len(vals)}
|
||
|
|
}
|
||
|
|
a, err := core.FromFloats(vals, shape...)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("FromFloats(%v, %v): %v", vals, shape, err)
|
||
|
|
}
|
||
|
|
return a
|
||
|
|
}
|