Moved test out of run_in_graph_and_eager_mode in dense_image_warp. (#1419)

See #1328

Author: gabrieldemarmiesse

tensorflow_addons/image/dense_image_warp_test.py

@@ -74,18 +74,20 @@ def test_interpolate_small_grid_batched(self):
 
         self.assertAllClose(expected_results, interp)
 
-    def test_unknown_shape(self):
-        query_points = tf.constant(
-            [[0.0, 0.0], [0.0, 1.0], [0.5, 2.0], [1.5, 1.5]], shape=[1, 4, 2]
-        )
-        fn = interpolate_bilinear.get_concrete_function(
-            tf.TensorSpec(shape=None, dtype=tf.float32),
-            tf.TensorSpec(shape=None, dtype=tf.float32),
-        )
-        for shape in (2, 4, 3, 6), (6, 2, 4, 3), (1, 2, 4, 3):
-            image = tf.ones(shape=shape)
-            res = fn(image, query_points)
-            self.assertAllEqual(res.shape, (shape[0], 4, shape[3]))
+
+@pytest.mark.usefixtures("maybe_run_functions_eagerly")
+def test_unknown_shape():
+    query_points = tf.constant(
+        [[0.0, 0.0], [0.0, 1.0], [0.5, 2.0], [1.5, 1.5]], shape=[1, 4, 2]
+    )
+    fn = interpolate_bilinear.get_concrete_function(
+        tf.TensorSpec(shape=None, dtype=tf.float32),
+        tf.TensorSpec(shape=None, dtype=tf.float32),
+    )
+    for shape in (2, 4, 3, 6), (6, 2, 4, 3), (1, 2, 4, 3):
+        image = tf.ones(shape=shape)
+        res = fn(image, query_points)
+        assert res.shape == (shape[0], 4, shape[3])
 
 
 @test_utils.run_all_in_graph_and_eager_modes