Add test for 2D integration