Pytest Function To Count Rows