单元测试返回矩阵的两个对象(java)

public class Matrix {
    private int[][] data;

    public Matrix(int[][] data) {
        this.data = data;
    }

    public Matrix add(Matrix other) {
        if (this.data.length != other.data.length || this.data[0].length != other.data[0].length) {
            throw new IllegalArgumentException("Matrices must have the same dimensions");
        }

        int[][] result = new int[this.data.length][this.data[0].length];
        for (int i = 0; i < this.data.length; i++) {
            for (int j = 0; j < this.data[0].length; j++) {
                result[i][j] = this.data[i][j] + other.data[i][j];
            }
        }
        return new Matrix(result);
    }

    public int[][] getData() {
        return data;
    }
}

import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.*;

public class MatrixTest {

    @Test
    public void testAdd() {
        int[][] data1 = {{1, 2}, {3, 4}};
        int[][] data2 = {{5, 6}, {7, 8}};
        Matrix matrix1 = new Matrix(data1);
        Matrix matrix2 = new Matrix(data2);

        Matrix result = matrix1.add(matrix2);
        int[][] expectedData = {{6, 8}, {10, 12}};
        assertArrayEquals(expectedData, result.getData());
    }

    @Test
    public void testAddDimensionMismatch() {
        int[][] data1 = {{1, 2}, {3, 4}};
        int[][] data2 = {{5, 6, 7}, {8, 9, 10}};
        Matrix matrix1 = new Matrix(data1);
        Matrix matrix2 = new Matrix(data2);

        assertThrows(IllegalArgumentException.class, () -> {
            matrix1.add(matrix2);
        });
    }
}