Update test_sgd.py

added rng functions