@@ -856,27 +856,27 @@ def test_random_normal(self):
856
856
mean = 0.
857
857
std = 1.
858
858
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
863
863
864
864
def test_random_uniform (self ):
865
865
min_val = - 1.
866
866
max_val = 1.
867
867
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
871
871
assert np .max (rand ) <= max_val
872
872
assert np .min (rand ) >= min_val
873
873
874
874
def test_random_binomial (self ):
875
875
p = 0.5
876
876
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
880
880
assert np .max (rand ) == 1
881
881
assert np .min (rand ) == 0
882
882
0 commit comments