Why Go
- Case Studies
  
  Common problems companies solve with Go
- Use Cases
  
  Stories about how and why companies use Go
- Security
  
  How Go can help keep you secure by default
Learn
Docs
- Effective Go
  
  Tips for writing clear, performant, and idiomatic Go code
- Go User Manual
  
  A complete introduction to building software with Go
- Standard library
  
  Reference documentation for Go's standard library
- Release Notes
  
  Learn what's new in each Go release
Packages
Community
- Recorded Talks
  
  Videos from prior events
- Meetups
  
  Meet other local Go developers
- Conferences
  
  Learn and network with Go developers from around the world
- Go blog
  
  The Go project's official blog.
- Go project
  
  Get help and stay informed from Go
- Get connected

testutils

package

Go to main page

Versions in this module

v1

v1.2.0

Mar 15, 2026

v1.1.0

Mar 14, 2026

v0

v0.3.0

Aug 25, 2025

v0.2.0

Aug 22, 2025

Changes in this version

+ func AssertFloat32SliceApproxEqual(t *testing.T, expected, actual []float32, tolerance float32, msg string)

+ func AssertFloat64SliceApproxEqual(t *testing.T, expected, actual []float64, tolerance float64, msg string)

+ func CompareTensorsApprox[T tensor.Numeric](t *testing.T, actual, expected *tensor.TensorNumeric[T], epsilon T) bool

+ func Int64SliceEqual(a, b []int64) bool

+ func RunTests(t *testing.T, tests []TestCase)

+ type FailingInitializer struct

+ func (f *FailingInitializer[T]) Initialize(inputSize, outputSize int) ([]T, error)

+ type TestCase struct

+ Func func(t *testing.T)

+ Name string

+ type TestInitializer struct

+ Value T

+ func (t *TestInitializer[T]) Initialize(inputSize, outputSize int) ([]T, error)

v0.1.0

Aug 4, 2025

Changes in this version

+ func AssertContains(t *testing.T, s, substr, msg string)

+ func AssertEqual[T comparable](t *testing.T, expected, actual T, msg string)

+ func AssertError(t *testing.T, err error, msg string)

+ func AssertFalse(t *testing.T, condition bool, msg string)

+ func AssertFloatEqual[T float32 | float64](t *testing.T, expected, actual T, tolerance T, msg string)

+ func AssertNil(t *testing.T, value interface{}, msg string)

+ func AssertNoError(t *testing.T, err error, msg string)

+ func AssertNotNil(t *testing.T, value interface{}, msg string)

+ func AssertPanics(t *testing.T, f func(), msg string)

+ func AssertTrue(t *testing.T, condition bool, msg string)

+ func ElementsMatch(a, b []string) bool

+ func IntSliceEqual(a, b []int) bool

+ func MockClientFactory(_ *grpc.ClientConn) pb.DistributedServiceClient

+ type CustomMockDistributedServiceClient struct

+ func (m *CustomMockDistributedServiceClient) AllReduce(_ context.Context, _ ...grpc.CallOption) (pb.DistributedService_AllReduceClient, error)

+ func (m *CustomMockDistributedServiceClient) AssertExpectations(t *testing.T)

+ func (m *CustomMockDistributedServiceClient) Barrier(_ context.Context, _ *pb.BarrierRequest, _ ...grpc.CallOption) (*pb.BarrierResponse, error)

+ func (m *CustomMockDistributedServiceClient) Broadcast(_ context.Context, _ *pb.BroadcastRequest, _ ...grpc.CallOption) (*pb.BroadcastResponse, error)

+ func (m *CustomMockDistributedServiceClient) OnAllReduce() *CustomMockDistributedServiceClient

+ func (m *CustomMockDistributedServiceClient) ReturnAllReduce(client pb.DistributedService_AllReduceClient, err error) *CustomMockDistributedServiceClient

+ type CustomMockGrpcServer struct

+ ServeErr error

+ func (m *CustomMockGrpcServer) AssertExpectations(t *testing.T)

+ func (m *CustomMockGrpcServer) GracefulStop()

+ func (m *CustomMockGrpcServer) OnServe(err error) *CustomMockGrpcServer

+ func (m *CustomMockGrpcServer) RegisterService(_ *grpc.ServiceDesc, _ interface{})

+ func (m *CustomMockGrpcServer) Serve(_ net.Listener) error

+ func (m *CustomMockGrpcServer) Stop()

+ type CustomMockListener struct

+ AcceptErr error

+ AddrVal net.Addr

+ CloseErr error

+ func (m *CustomMockListener) Accept() (net.Conn, error)

+ func (m *CustomMockListener) Addr() net.Addr

+ func (m *CustomMockListener) AssertExpectations(t *testing.T)

+ func (m *CustomMockListener) Close() error

+ func (m *CustomMockListener) OnAccept(err error) *CustomMockListener

+ func (m *CustomMockListener) OnAddr(addr net.Addr) *CustomMockListener

+ func (m *CustomMockListener) OnClose(err error) *CustomMockListener

+ type CustomMockLogger struct

+ func (m *CustomMockLogger) AssertExpectations(t *testing.T)

+ func (m *CustomMockLogger) OnPrintf() *CustomMockLogger

+ func (m *CustomMockLogger) Printf(format string, v ...interface{})

+ type CustomMockStrategy struct

+ BroadcastTensorArgs []struct{ ... }

+ func (m *CustomMockStrategy[T]) AllReduceGradients(gradients map[string]*tensor.Tensor[T]) error

+ func (m *CustomMockStrategy[T]) AssertExpectations(t *testing.T)

+ func (m *CustomMockStrategy[T]) AssertNotCalled(t *testing.T, methodName string)

+ func (m *CustomMockStrategy[T]) Barrier() error

+ func (m *CustomMockStrategy[T]) BroadcastTensor(t *tensor.Tensor[T], rootRank int) error

+ func (m *CustomMockStrategy[T]) Init(rank int, size int, coordinatorAddress string) error

+ func (m *CustomMockStrategy[T]) OnAllReduceGradients(gradients map[string]*tensor.Tensor[T]) *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) OnBarrier() *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) OnBroadcastTensor(t *tensor.Tensor[T], rootRank int) *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) OnInit(rank, size int, coordinatorAddress string) *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) OnRank() *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) OnSize() *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) OnceAllReduceGradients() *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) OnceBroadcastTensor() *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) OnceInit() *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) OnceRank() *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) OnceSize() *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) Rank() int

+ func (m *CustomMockStrategy[T]) ReturnAllReduceGradients(err error) *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) ReturnBarrier(err error) *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) ReturnBroadcastTensor(err error) *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) ReturnInit(err error) *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) ReturnRank(rank int) *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) ReturnSize(size int) *CustomMockStrategy[T]

+ func (m *CustomMockStrategy[T]) Shutdown()

+ func (m *CustomMockStrategy[T]) Size() int

+ func (m *CustomMockStrategy[T]) TwiceBarrier() *CustomMockStrategy[T]

+ type MockEngine struct

+ Err error

+ func NewMockEngineWithError[T tensor.Numeric](err error) *MockEngine[T]

+ func NewMockEngine[T tensor.Numeric]() *MockEngine[T]

+ func (e *MockEngine[T]) Add(_ context.Context, _, _ *tensor.Tensor[T], _ ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (e *MockEngine[T]) Allocator() device.Allocator

+ func (e *MockEngine[T]) Close() error

+ func (e *MockEngine[T]) Context() context.Context

+ func (e *MockEngine[T]) Copy(_ context.Context, _, _ *tensor.Tensor[T]) error

+ func (e *MockEngine[T]) Device() device.Device

+ func (e *MockEngine[T]) Div(_ context.Context, _, _ *tensor.Tensor[T], _ ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (e *MockEngine[T]) Exp(_ context.Context, _ *tensor.Tensor[T], _ ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (e *MockEngine[T]) Log(_ context.Context, _ *tensor.Tensor[T], _ ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (e *MockEngine[T]) MatMul(_ context.Context, _, _ *tensor.Tensor[T], _ ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (e *MockEngine[T]) Mul(_ context.Context, _, _ *tensor.Tensor[T], _ ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (e *MockEngine[T]) Name() string

+ func (e *MockEngine[T]) Pow(_ context.Context, _, _ *tensor.Tensor[T], _ ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (e *MockEngine[T]) String() string

+ func (e *MockEngine[T]) Sub(_ context.Context, _, _ *tensor.Tensor[T], _ ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (e *MockEngine[T]) Sum(_ context.Context, _ *tensor.Tensor[T], _ int, _ bool, _ ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (e *MockEngine[T]) Transpose(_ context.Context, _ *tensor.Tensor[T], _ []int, _ ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (e *MockEngine[T]) UnaryOp(_ context.Context, _ *tensor.Tensor[T], _ func(T) T, _ ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (e *MockEngine[T]) Wait() error

+ func (e *MockEngine[T]) WithAllocator(_ device.Allocator) compute.Engine[T]

+ func (e *MockEngine[T]) WithContext(_ context.Context) compute.Engine[T]

+ func (e *MockEngine[T]) WithDevice(_ device.Device) compute.Engine[T]

+ func (e *MockEngine[T]) WithError(err error) *MockEngine[T]

+ func (e *MockEngine[T]) WithName(_ string) compute.Engine[T]

+ func (e *MockEngine[T]) Zero(_ context.Context, _ *tensor.Tensor[T]) error