Refactor merging layers tests as parametrized tests. (#913)

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.