Modified testCommentCount function in tests.py