Added basic tests for expected row counts