Add expect method for checking count exactly