@@ -856,27 +856,27 @@ def test_random_normal(self):
856856 mean = 0.
857857 std = 1.
858858 for k in BACKENDS :
859- rand = k .eval (k .random_normal ((1000 , 1000 ), mean = mean , stddev = std ))
860- assert rand .shape == (1000 , 1000 )
861- assert np .abs (np .mean (rand ) - mean ) < 0.01
862- assert np .abs (np .std (rand ) - std ) < 0.01
859+ rand = k .eval (k .random_normal ((200 , 100 ), mean = mean , stddev = std ))
860+ assert rand .shape == (200 , 100 )
861+ assert np .abs (np .mean (rand ) - mean ) < 0.015
862+ assert np .abs (np .std (rand ) - std ) < 0.015
863863
864864 def test_random_uniform (self ):
865865 min_val = - 1.
866866 max_val = 1.
867867 for k in BACKENDS :
868- rand = k .eval (k .random_uniform ((1000 , 1000 ), min_val , max_val ))
869- assert rand .shape == (1000 , 1000 )
870- assert np .abs (np .mean (rand )) < 0.01
868+ rand = k .eval (k .random_uniform ((200 , 100 ), min_val , max_val ))
869+ assert rand .shape == (200 , 100 )
870+ assert np .abs (np .mean (rand )) < 0.015
871871 assert np .max (rand ) <= max_val
872872 assert np .min (rand ) >= min_val
873873
874874 def test_random_binomial (self ):
875875 p = 0.5
876876 for k in BACKENDS :
877- rand = k .eval (k .random_binomial ((1000 , 1000 ), p ))
878- assert rand .shape == (1000 , 1000 )
879- assert np .abs (np .mean (rand ) - p ) < 0.01
877+ rand = k .eval (k .random_binomial ((200 , 100 ), p ))
878+ assert rand .shape == (200 , 100 )
879+ assert np .abs (np .mean (rand ) - p ) < 0.015
880880 assert np .max (rand ) == 1
881881 assert np .min (rand ) == 0
882882
0 commit comments