Testing Utiltites

xfail_if_not_implemented(msg='Not implemented', *, match=None)

xfail_if_not_found(msg='Not implemented')

requires_backend(*backends, reason=None)

excludes_backend(*backends, reason=None)

class ActualExpected

Bases: funsor.testing.LazyComparison

Lazy string formatter for test assertions.

id_from_inputs(inputs)

is_array(x)

assert_close(actual, expected, atol=1e-06, rtol=1e-06)

check_funsor(x, inputs, output, data=None)

Check dims and shape modulo reordering.

xfail_param(*args, **kwargs)

make_einsum_example(equation, fill=None, sizes=(2, 3))

assert_equiv(x, y)

Check that two funsors are equivalent up to permutation of inputs.

rand(*args)

randint(low, high, size)

randn(*args)

random_scale_tril(*args)

zeros(*args)

ones(*args)

empty(*args)

random_tensor(inputs, output=Real)

Creates a random funsor.tensor.Tensor with given inputs and output.

random_gaussian(inputs)

Creates a random funsor.gaussian.Gaussian with given inputs.

random_mvn(batch_shape, dim, diag=False)

Generate a random torch.distributions.MultivariateNormal with given shape.

make_plated_hmm_einsum(num_steps, num_obs_plates=1, num_hidden_plates=0)

make_chain_einsum(num_steps)

make_hmm_einsum(num_steps)

iter_subsets(iterable, *, min_size=None, max_size=None)

class DesugarGetitem

Bases: object

Helper to desugar .__getitem__() syntax.

Example:

>>> desugar_getitem[1:3, ..., None]
(slice(1, 3), Ellipsis, None)