graphtest

Documentation

Overview

Package graphtest holds test utilities for packages that depend on the graph package.

Index

• func BuildTestBackend() backends.Backend
• func RunTestGraphFn(t *testing.T, testName string, graphFn TestGraphFn, want []any, delta float64)
• type TestGraphFn

Functions

func BuildTestBackend

func BuildTestBackend() backends.Backend

BuildTestBackend and sets backends.DefaultConfig to "xla:cpu" -- it can be overwritten by GOMLX_BACKEND environment variable.

func RunTestGraphFn

func RunTestGraphFn(t *testing.T, testName string, graphFn TestGraphFn, want []any, delta float64)

RunTestGraphFn tests a graph building function graphFn by executing it and comparing its output(s) to the values in want, reporting back any errors in t.

delta is the margin of value on the difference of output and want values that are acceptable. Values of delta <= 0 means only exact equality is accepted.

Types

type TestGraphFn

type TestGraphFn func(g *graph.Graph) (inputs, outputs []*graph.Node)

TestGraphFn should build its own inputs, and return both inputs and outputs