def test_shape(self, batch_size, device, dtype):
B: int = batch_size
focal: float = 100.0
image = torch.rand(B, 3, 4, 4, device=device, dtype=dtype)
camera_matrix = epi.intrinsics_like(focal, image)
assert camera_matrix.shape == (B, 3, 3)
assert camera_matrix.device == image.device
assert camera_matrix.dtype == image.dtype
def test_smoke(self, device, dtype):
image = torch.rand(1, 3, 4, 4, device=device, dtype=dtype)
focal = torch.rand(1, device=device, dtype=dtype)
camera_matrix = epi.intrinsics_like(focal, image)
assert camera_matrix.shape == (1, 3, 3)