Add resnet50 test.

satels · Mar 3, 2017 · 5b0967a · 5b0967a
1 parent 136f1b4
commit 5b0967a
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tests/keras/applications/applications_test.py b/tests/keras/applications/applications_test.py
@@ -4,6 +4,26 @@
 from keras import backend as K
 
 
+@keras_test
+def test_resnet50():
+    model = applications.ResNet50(weights=None)
+    assert model.output_shape == (None, 1000)
+
+
+@keras_test
+def test_resnet50_notop():
+    model = applications.ResNet50(weights=None, include_top=False)
+    assert model.output_shape == (None, None, None, 2048)
+
+
+@keras_test
+def test_resnet50_pooling():
+    model = applications.ResNet50(weights=None,
+                                  include_top=False,
+                                  pooling='avg')
+    assert model.output_shape == (None, 2048)
+
+
 @keras_test
 def test_vgg16():
     model = applications.VGG16(weights=None)