PEP8 fixes in tests.

This commit is contained in:
Francois Chollet 2017-01-11 11:40:57 -08:00
parent 309f586424
commit c10945f53a
8 changed files with 20 additions and 14 deletions

@ -108,8 +108,8 @@ def test_stacked_lstm_char_prediction():
y = np.zeros((len(sentences), number_of_chars), dtype=np.bool)
for i, sentence in enumerate(sentences):
for t, char in enumerate(sentence):
X[i, t, ord(char)-ord('a')] = 1
y[i, ord(next_chars[i])-ord('a')] = 1
X[i, t, ord(char) - ord('a')] = 1
y[i, ord(next_chars[i]) - ord('a')] = 1
# learn the alphabet with stacked LSTM
model = Sequential([
@ -123,7 +123,7 @@ def test_stacked_lstm_char_prediction():
# prime the model with 'ab' sequence and let it generate the learned alphabet
sentence = alphabet[:sequence_length]
generated = sentence
for iteration in range(number_of_chars-sequence_length):
for iteration in range(number_of_chars - sequence_length):
x = np.zeros((1, sequence_length, number_of_chars))
for t, char in enumerate(sentence):
x[0, t, ord(char) - ord('a')] = 1.

@ -899,7 +899,7 @@ class TestBackend(object):
def test_foldl(self):
x = np.random.rand(10, 3).astype(np.float32)
for K in [KTF, KTH]:
kx = K.eval(K.foldl(lambda a, b: a+b, x))
kx = K.eval(K.foldl(lambda a, b: a + b, x))
assert (3,) == kx.shape
assert_allclose(x.sum(axis=0), kx, atol=1e-05)
@ -911,8 +911,8 @@ class TestBackend(object):
# right to left we have no such problem and the result is larger
x = np.array([1e-20, 1e-20, 10, 10, 10], dtype=np.float32)
for K in [KTF, KTH]:
p1 = K.eval(K.foldl(lambda a, b: a*b, x))
p2 = K.eval(K.foldr(lambda a, b: a*b, x))
p1 = K.eval(K.foldl(lambda a, b: a * b, x))
p2 = K.eval(K.foldr(lambda a, b: a * b, x))
assert p1 < p2
assert 9e-38 < p2 <= 1e-37

@ -9,6 +9,7 @@ from keras import backend as K
from keras.models import model_from_json, model_from_yaml
from keras.utils.test_utils import keras_test
@keras_test
def test_get_updates_for():
a = Input(shape=(2,))

@ -15,7 +15,7 @@ class TestImage:
gray_images = []
for n in range(8):
bias = np.random.rand(img_w, img_h, 1) * 64
variance = np.random.rand(img_w, img_h, 1) * (255-64)
variance = np.random.rand(img_w, img_h, 1) * (255 - 64)
imarray = np.random.rand(img_w, img_h, 3) * variance + bias
im = Image.fromarray(imarray.astype('uint8')).convert('RGB')
rgb_images.append(im)

@ -96,6 +96,7 @@ def test_ModelCheckpoint():
os.remove(filepath.format(epoch=1))
os.remove(filepath.format(epoch=3))
def test_EarlyStopping():
(X_train, y_train), (X_test, y_test) = get_test_data(nb_train=train_samples,
nb_test=test_samples,

@ -86,10 +86,10 @@ def test_identity(tensor_shape):
if len(tensor_shape) > 2:
with pytest.raises(Exception):
_runner(initializations.identity, tensor_shape,
target_mean=1./SHAPE[0], target_max=1.)
target_mean=1. / SHAPE[0], target_max=1.)
else:
_runner(initializations.identity, tensor_shape,
target_mean=1./SHAPE[0], target_max=1.)
target_mean=1. / SHAPE[0], target_max=1.)
@pytest.mark.parametrize('tensor_shape', [FC_SHAPE, CONV_SHAPE], ids=['FC', 'CONV'])

@ -49,6 +49,7 @@ def test_clasify_build_fn():
def test_clasify_class_build_fn():
class ClassBuildFnClf(object):
def __call__(self, hidden_dims):
return build_fn_clf(hidden_dims)
@ -61,6 +62,7 @@ def test_clasify_class_build_fn():
def test_clasify_inherit_class_build_fn():
class InheritClassBuildFnClf(KerasClassifier):
def __call__(self, hidden_dims):
return build_fn_clf(hidden_dims)
@ -110,6 +112,7 @@ def test_regression_build_fn():
def test_regression_class_build_fn():
class ClassBuildFnReg(object):
def __call__(self, hidden_dims):
return build_fn_reg(hidden_dims)
@ -122,6 +125,7 @@ def test_regression_class_build_fn():
def test_regression_inherit_class_build_fn():
class InheritClassBuildFnReg(KerasRegressor):
def __call__(self, hidden_dims):
return build_fn_reg(hidden_dims)