pass in count and num loops to unit test to make it shorter