pose_format.torch.masked.torch_test module

Classes:

TestMaskedTorch([methodName])

Test cases for the MaskedTensor class

class pose_format.torch.masked.torch_test.TestMaskedTorch(methodName='runTest')[source]

Bases: TestCase

Test cases for the MaskedTensor class

Methods:

test_cat()

Test cat method for concatenating MaskedTensor objects along a specified dimension.

test_not_implemented_method()

Tests behavior when invoking an unimplemented method on a MaskedTensor.

test_stack()

Tests stack method for stacking MaskedTensor objects along a new dimension.

test_zeros_mask_value()

Test if the mask in the produced zeros MaskedTensor is initialized with zero values.

test_zeros_tensor_shape()

Test if zeros method correctly produces a MaskedTensor with the desired shape.

test_zeros_tensor_type_bool()

Test if the zeros method produces a MaskedTensor with the correct boolean data type.

test_zeros_tensor_type_float()

Test if the zeros method produces a MaskedTensor with the correct float data type.

test_zeros_tensor_value()

Test if the zeros method produces a MaskedTensor with all zero values.

test_cat()[source]

Test cat method for concatenating MaskedTensor objects along a specified dimension.

test_not_implemented_method()[source]

Tests behavior when invoking an unimplemented method on a MaskedTensor.

test_stack()[source]

Tests stack method for stacking MaskedTensor objects along a new dimension.

test_zeros_mask_value()[source]

Test if the mask in the produced zeros MaskedTensor is initialized with zero values.

test_zeros_tensor_shape()[source]

Test if zeros method correctly produces a MaskedTensor with the desired shape.

test_zeros_tensor_type_bool()[source]

Test if the zeros method produces a MaskedTensor with the correct boolean data type.

test_zeros_tensor_type_float()[source]

Test if the zeros method produces a MaskedTensor with the correct float data type.

test_zeros_tensor_value()[source]

Test if the zeros method produces a MaskedTensor with all zero values.