Add test

apache · Mar 25, 2019 · fea2f1a · fea2f1a
1 parent a744cc3
commit fea2f1a
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/python/frontend/keras/test_forward.py b/tests/python/frontend/keras/test_forward.py
@@ -106,6 +106,17 @@ def test_forward_dense():
     verify_keras_frontend(keras_model)
 
 
+def test_forward_sequential():
+    keras_model = keras.models.Sequential([
+        keras.layers.Dense(16, input_dim=32, activation='relu'),
+        keras.layers.Dropout(0.5),
+        keras.layers.Dense(8, activation='relu'),
+        keras.layers.Dropout(0.5),
+        keras.layers.Dense(1, activation='sigmoid')
+    ])
+    verify_keras_frontend(keras_model)
+
+
 def test_forward_pool():
     data = keras.layers.Input(shape=(32,32,1))
     # maxpool
@@ -244,6 +255,7 @@ def test_forward_mobilenet():
     test_forward_merge()
     test_forward_activations()
     test_forward_dense()
+    test_forward_sequential()
     test_forward_pool()
     test_forward_conv()
     test_forward_upsample(interpolation='nearest')