def test_zeros_mask_value(self):
zeros = MaskedTorch.zeros(1, 2, 3)
self.assertTrue(torch.all(zeros.mask == 0),
msg="Zeros mask are not all zeros")
def test_zeros_tensor_value(self):
zeros = MaskedTorch.zeros(1, 2, 3)
self.assertTrue(torch.all(zeros == 0), msg="Zeros are not all zeros")
def test_zeros_tensor_type_bool(self):
dtype = torch.bool
zeros = MaskedTorch.zeros(1, 2, 3, dtype=dtype)
self.assertEqual(zeros.tensor.dtype, dtype)
def test_zeros_tensor_shape(self):
zeros = MaskedTorch.zeros(1, 2, 3)
self.assertEqual(zeros.shape, (1, 2, 3))