Refactor merging layers tests as parametrized tests.

They are almost identical for every merging layer.

This simplifies adding tests that apply to every layer. This is in preparation for adding sparse tensors support.