Write tests for 1D and 3D