Update test_train.py

added rng_randn