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

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

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

@ -49,6 +49,7 @@ def test_clasify_build_fn():
def test_clasify_class_build_fn(): def test_clasify_class_build_fn():
class ClassBuildFnClf(object): class ClassBuildFnClf(object):
def __call__(self, hidden_dims): def __call__(self, hidden_dims):
return build_fn_clf(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(): def test_clasify_inherit_class_build_fn():
class InheritClassBuildFnClf(KerasClassifier): class InheritClassBuildFnClf(KerasClassifier):
def __call__(self, hidden_dims): def __call__(self, hidden_dims):
return build_fn_clf(hidden_dims) return build_fn_clf(hidden_dims)
@ -110,6 +112,7 @@ def test_regression_build_fn():
def test_regression_class_build_fn(): def test_regression_class_build_fn():
class ClassBuildFnReg(object): class ClassBuildFnReg(object):
def __call__(self, hidden_dims): def __call__(self, hidden_dims):
return build_fn_reg(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(): def test_regression_inherit_class_build_fn():
class InheritClassBuildFnReg(KerasRegressor): class InheritClassBuildFnReg(KerasRegressor):
def __call__(self, hidden_dims): def __call__(self, hidden_dims):
return build_fn_reg(hidden_dims) return build_fn_reg(hidden_dims)