abracudabra.testing

Assertions for testing purposes.

Functions

assert_tensors_equal(→ None)	Assert that two torch tensors are equal.
assert_arrays_equal(→ None)	Assert that two arrays are equal.
assert_frames_equal(→ None)	Assert that two pandas or cudf series or dataframes are equal.

Module Contents

abracudabra.testing.assert_tensors_equal(tensor1: torch.Tensor, tensor2: torch.Tensor, /) → None

Assert that two torch tensors are equal.

abracudabra.testing.assert_arrays_equal(array1: abracudabra.annotations.Array, array2: abracudabra.annotations.Array, /) → None

Assert that two arrays are equal.

abracudabra.testing.assert_frames_equal(frame1: abracudabra.annotations.Series | abracudabra.annotations.DataFrame, frame2: abracudabra.annotations.Series | abracudabra.annotations.DataFrame) → None

Assert that two pandas or cudf series or dataframes are equal.