Update tests/stochastic/test.py