Minor update taskcoach_test.py
Reorganized taskcoach_test.py for better readability