test: add a test for 2d arrays