From 42fcf5554a1813781c61a2c1de0665822e86b024 Mon Sep 17 00:00:00 2001
From: Piv <18462828+Piv200@users.noreply.github.com>
Date: Tue, 24 Aug 2021 21:39:10 +0930
Subject: [PATCH] Update warp test to verify output shape

---
 unsupervised/warp_tests.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/unsupervised/warp_tests.py b/unsupervised/warp_tests.py
index 626c7e1..1cf62a1 100644
--- 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__':