fix selected module in mobilenet test : it should be dependant on sel…

…ected mobilenet version (#25)
keras-team · Aug 3, 2018 · fb46b9a · fb46b9a
1 parent 8ad3241
commit fb46b9a
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/applications_test.py b/tests/applications_test.py
@@ -30,8 +30,8 @@
 from multiprocessing import Process, Queue
 
 
-MOBILENET_LIST = [(mobilenet.MobileNet, 1024),
-                  (mobilenet_v2.MobileNetV2, 1280)]
+MOBILENET_LIST = [(mobilenet.MobileNet, mobilenet, 1024),
+                  (mobilenet_v2.MobileNetV2, mobilenet_v2, 1280)]
 DENSENET_LIST = [(densenet.DenseNet121, 1024),
                  (densenet.DenseNet169, 1664),
                  (densenet.DenseNet201, 1920)]
@@ -190,8 +190,7 @@ def test_inceptionresnetv2():
 
 
 def test_mobilenet():
-    app, last_dim = random.choice(MOBILENET_LIST)
-    module = mobilenet
+    app, module, last_dim = random.choice(MOBILENET_LIST)
     _test_application_basic(app, module=module)
     _test_application_notop(app, last_dim)
     _test_application_variable_input_channels(app, last_dim)