[MEGA] nit size test
# What does this PR do?
Adresses #23025, the input_shape should be tested, not input ids because it might be None.
_The documentation is not available anymore as the PR was closed or merged._