Fixed test name and added a further test with custom Model and Conv1D layer.

Author: roebel

keras/src/layers/reshaping/reshape_test.py

@@ -1,6 +1,7 @@
 import pytest
 from absl.testing import parameterized
 
+from keras.src import Model
 from keras.src import ops
 from keras.src import backend
 from keras.src import layers
@@ -101,7 +102,7 @@ def test_reshape_with_dynamic_batch_size_and_minus_one(self):
         reshaped = backend.compute_output_spec(layer.__call__, input)
         self.assertEqual(reshaped.shape, (None, 3, 8))
 
-    def test_reshape_with_varying_static_batch_size_and_minus_one(self):
+    def test_reshape_layer_with_varying_input_size_and_minus_one(self):
         input = KerasTensor((None, 6, 4))
         layer = layers.Reshape((-1, 8))
         layer.build(input.shape)
@@ -110,6 +111,23 @@ def test_reshape_with_varying_static_batch_size_and_minus_one(self):
         res = layer(ops.ones((1, 10, 4), dtype="float32"))
         self.assertEqual(res.shape, (1, 5, 8))
 
+    def test_custom_reshape_model_with_varying_input_size_and_minus_one(self):
+        class MM(layers.Layer):
+            def __init__(self):
+                super().__init__()
+                self.conv = layers.Conv1D(4, 3, padding="same")
+                self.reshape = layers.Reshape((-1, 8))
+
+            def call(self, inputs):
+                x = self.conv(inputs)
+                return self.reshape(x)
+
+        m = MM()
+        res = m(ops.ones((1, 6, 2), dtype="float32"))
+        self.assertEqual(res.shape, (1, 3, 8))
+        res = m(ops.ones((1, 10, 2), dtype="float32"))
+        self.assertEqual(res.shape, (1, 5, 8))
+
     def test_reshape_with_dynamic_dim_and_minus_one(self):
         input = KerasTensor((4, 6, None, 3))
         layer = layers.Reshape((-1, 3))