minor progress on test_model.py
Work done on gradient test