Add count line unit tests