Add more concatenate tests

Author: ValerianRey

tests/unit/sparse/test_diagonal_sparse_tensor.py

@@ -261,6 +261,10 @@ def test_unsquash_pdim(
 @mark.parametrize(
     ["dst_args", "dim"],
     [
+        ([([3], [[0], [0]]), ([3], [[0], [0]])], 1),
+        ([([3, 2], [[0], [0, 1]]), ([3], [[0], [0]])], 1),
+        ([([3], [[0], [0]]), ([3, 2], [[0], [0, 1]])], 1),
+        ([([3, 2], [[0], [0, 1]]), ([3, 2], [[0], [0, 1]])], 1),
         ([([3, 4], [[0], [0, 1]]), ([3, 3, 4], [[0, 1], [1, 2]])], 0),
         ([([3, 12], [[0, 1], [0]]), ([9, 4], [[0, 1], [0]])], 1),
     ],