Add a unit test for embedded 2-dimensional for loop