Update test_probabilistic_max_pooling.py

added rng functions