Adding some comments to test_chi.py

I have added some suitable comments into the code.