add test_grade.py