added tests for one cell and two cells