[src][test] Adding in tests and implementation for getting rows and columns