Update warp test to verify output shape

2021-08-24 21:39:10 +09:30
parent c164c9720a
commit 42fcf5554a
1 changed files with 1 additions and 1 deletions
--- a/unsupervised/warp_tests.py
+++ b/unsupervised/warp_tests.py
@@ -56,7 +56,7 @@ class MyTestCase(unittest.TestCase):
        disp = tf.random.uniform([1, height, width]) * 255
        pose = tf.random.uniform([1, 6])

-        warp.projective_inverse_warp(img, disp, pose, intrinsics, coords)
+        self.assertEqual(warp.projective_inverse_warp(img, disp, pose, intrinsics, coords).shape, img.shape)


 if __name__ == '__main__':