added test on test_sgd.py