Added documentation in test.py