Revise unit test

1 job for develop-merge_inner_dim_with_diff_lengths
in 58 minutes and 46 seconds, using 0 compute credits, and was queued for 3 seconds