Using test case JUnit 5 to compare 2 2D arrays

3 min read 05-10-2024
Using test case JUnit 5 to compare 2 2D arrays


Comparing 2D Arrays with JUnit 5: A Guide for Developers

Testing is a crucial part of software development, ensuring your code behaves as expected. When working with data structures like 2D arrays, comparing their contents accurately is vital. This article explores how to utilize JUnit 5's powerful assertion capabilities to compare 2D arrays effectively.

The Challenge: Comparing 2D Arrays

Imagine you have a function that manipulates a 2D array, like sorting its elements or performing matrix operations. To validate its correctness, you need to compare the expected output (the expected 2D array) with the actual output produced by your function. This is where JUnit 5 comes in.

A Simple Example:

Let's consider a function that reverses the rows of a 2D array:

public static int[][] reverseRows(int[][] inputArray) {
    int[][] reversedArray = new int[inputArray.length][inputArray[0].length];
    for (int i = 0; i < inputArray.length; i++) {
        for (int j = 0; j < inputArray[0].length; j++) {
            reversedArray[i][j] = inputArray[i][inputArray[0].length - 1 - j];
        }
    }
    return reversedArray;
}

To test this function with JUnit 5, we can write the following test case:

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

public class ArrayReversalTest {

    @Test
    void testReverseRows() {
        int[][] inputArray = {{1, 2, 3}, {4, 5, 6}};
        int[][] expectedArray = {{3, 2, 1}, {6, 5, 4}};
        int[][] actualArray = reverseRows(inputArray);

        assertArrayEquals(expectedArray, actualArray);
    }
}

Explanation:

  • We use the assertArrayEquals() method from JUnit 5's Assertions class to compare the expected 2D array (expectedArray) with the actual 2D array (actualArray) returned by our function.
  • If the arrays are identical, the test passes. Otherwise, it fails, providing informative error messages.

Beyond Basic Comparisons

While assertArrayEquals() is a great starting point, it's essential to understand its limitations. It performs a shallow comparison, only checking if the array references are equal. This means that if the arrays contain nested arrays (like our example above), it will only compare the references of the inner arrays, not their content.

To handle nested arrays, we need to introduce a custom comparison mechanism:

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

public class ArrayReversalTest {

    @Test
    void testReverseRows() {
        int[][] inputArray = {{1, 2, 3}, {4, 5, 6}};
        int[][] expectedArray = {{3, 2, 1}, {6, 5, 4}};
        int[][] actualArray = reverseRows(inputArray);

        // Compare each row individually
        for (int i = 0; i < inputArray.length; i++) {
            assertArrayEquals(expectedArray[i], actualArray[i], "Row " + i + " does not match.");
        }
    }
}

Explanation:

  • We iterate through each row of the arrays.
  • For each row, we use assertArrayEquals() to compare the individual elements of the corresponding rows.
  • This ensures a deep comparison, checking the contents of each nested array within the 2D arrays.

Key Takeaways:

  • JUnit 5's assertArrayEquals() method is useful for comparing basic 2D arrays, but it only performs a shallow comparison.
  • To compare nested arrays, you need to implement a custom comparison logic using nested loops and the assertArrayEquals() method for the individual rows.
  • By writing comprehensive JUnit 5 tests, you can ensure your code is reliable and handles 2D array manipulation correctly.

Additional Resources:

By utilizing JUnit 5's tools and understanding how to compare 2D arrays effectively, you can confidently write robust and reliable software.