Enable integration tests (#9)

keras-team · Jul 28, 2018 · c457429 · c457429
1 parent 42f3e52
commit c457429
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/tests/integration_tests.py b/tests/integration_tests.py
@@ -16,17 +16,13 @@ def test_that_internal_imports_are_not_overriden():
     if not hasattr(keras.applications, 'keras_applications'):
         return  # Old Keras, don't run.
 
-    # TODO: enable the following tests
-    # if get_source_inputs is included in tf.keras.utils
-    return
-
     import tensorflow as tf
     keras_applications.set_keras_submodules(
         backend=tf.keras.backend,
         layers=tf.keras.layers,
         models=tf.keras.models,
         utils=tf.keras.utils)
-    assert keras.applications.vgg16.backend is keras.backend
+    assert keras_applications._KERAS_BACKEND.__name__ == 'tensorflow.keras.backend'
 
     # Now test the reverse order
     del keras
@@ -39,7 +35,7 @@ def test_that_internal_imports_are_not_overriden():
         models=tf.keras.models,
         utils=tf.keras.utils)
     import keras
-    assert keras.applications.vgg16.backend is keras.backend
+    assert keras_applications._KERAS_BACKEND.__name__ == 'tensorflow.keras.backend'
 
 
 if __name__ == '__main__':