def test_wrong_shape(self):
mats = [
np.ones((2, 3, 4)),
np.ones((3, 3))
]
with self.assertRaises(tf.errors.InvalidArgumentError):
for m in mats:
self.sess.run(tri_to_vec(m))
def testTriToVec(self):
mats = [
np.arange(1, 4)[None, :],
np.vstack((np.arange(1, 4), np.arange(3, 0, -1))), # Currently, only do matrices
np.arange(1, 16)[None, :]
]
with tf.Session(''):
for m in mats:
# The ops are each others' inverse.
self.assertTrue(np.all(tri_to_vec(vec_to_tri(m)).eval() == m))
