Added comment unit tests