Start adding pose warp conversions
This commit is contained in:
26
unsupervised/warp_tests.py
Normal file
26
unsupervised/warp_tests.py
Normal file
@@ -0,0 +1,26 @@
|
||||
import unittest
|
||||
|
||||
import numpy as np
|
||||
import tensorflow as tf
|
||||
|
||||
import warp
|
||||
|
||||
|
||||
class MyTestCase(unittest.TestCase):
|
||||
def test_euler_to_rotation_matrix(self):
|
||||
# quarter rotation in every
|
||||
x, y, z = tf.expand_dims(tf.expand_dims(tf.constant(np.pi / 2), 0), 0)
|
||||
|
||||
# TODO: Construct expected final rotation matrix, just 3x3 using numpy, so that we can do an
|
||||
# elementwise comparison later. Probably also want to check the
|
||||
|
||||
rotation_matrices = warp.euler_to_rotation_matrix(x, y, z)
|
||||
|
||||
self.assertEqual(rotation_matrices.shape, [1, 3, 3])
|
||||
rot_mat = rotation_matrices[0]
|
||||
|
||||
# TODO: Element-wise checks...
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user