a828315185a9dc8b21ec8e5dbead9044caf0d3a2,test/geometry/test_conversions.py,TestRotationMatrixToQuaternion,test_smoke_batch,#TestRotationMatrixToQuaternion#Any#Any#,62
Before Change
@pytest.mark.parametrize("batch_size", (1, 3, 8))
def test_smoke_batch(self, device, batch_size):
matrix = torch.zeros(batch_size, 3, 3).to(device)
quaternion = kornia.rotation_matrix_to_quaternion(matrix)
assert quaternion.shape == (batch_size, 4)
def test_identity(self, device):
matrix = torch.tensor([
After Change
quaternion = kornia.angle_axis_to_quaternion(angle_axis)
assert quaternion.shape == (4,)
@pytest.mark.parametrize("batch_size", (1, 3, 8))
def test_smoke_batch(self, batch_size, device, dtype):
angle_axis = torch.zeros(batch_size, 3, device=device, dtype=dtype)
quaternion = kornia.angle_axis_to_quaternion(angle_axis)
assert quaternion.shape == (batch_size, 4)
def test_zero_angle(self, device, dtype):
In pattern: SUPERPATTERN
Frequency: 3
Non-data size: 5
Instances
Project Name: arraiy/torchgeometry
Commit Name: a828315185a9dc8b21ec8e5dbead9044caf0d3a2
Time: 2020-12-22
Author: sj8716643@126.com
File Name: test/geometry/test_conversions.py
Class Name: TestRotationMatrixToQuaternion
Method Name: test_smoke_batch
Project Name: arraiy/torchgeometry
Commit Name: a828315185a9dc8b21ec8e5dbead9044caf0d3a2
Time: 2020-12-22
Author: sj8716643@126.com
File Name: test/geometry/test_conversions.py
Class Name: TestAngleAxisToQuaternion
Method Name: test_smoke_batch
Project Name: arraiy/torchgeometry
Commit Name: a828315185a9dc8b21ec8e5dbead9044caf0d3a2
Time: 2020-12-22
Author: sj8716643@126.com
File Name: test/geometry/test_linalg.py
Class Name: TestInverseTransformation
Method Name: test_gradcheck