Skip to content

Commit bfa7dca

Browse files
committed
Speed up backend tests
1 parent 7fc5466 commit bfa7dca

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

tests/keras/backend/backend_test.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)