fix(//tests/modules): Remove lenet test and rename generated

mobilenet_v2 module in hub.py

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

tests/modules/hub.py

@@ -10,7 +10,7 @@
           "inception": models.inception_v3(pretrained=True),
           "googlenet": models.googlenet(pretrained=True),
           "shufflenet": models.shufflenet_v2_x1_0(pretrained=True),
-          "mobilenet": models.mobilenet_v2(pretrained=True),
+          "mobilenet_v2": models.mobilenet_v2(pretrained=True),
           "resnext50_32x4d": models.resnext50_32x4d(pretrained=True),
           "wideresnet50_2": models.wide_resnet50_2(pretrained=True),
           "mnasnet": models.mnasnet1_0(pretrained=True),

tests/modules/test_compiled_modules.cpp

@@ -28,11 +28,9 @@ TEST_P(ModuleTests, CompiledModuleIsClose) {
 INSTANTIATE_TEST_SUITE_P(CompiledModuleForwardIsCloseSuite,
                          ModuleTests,
                          testing::Values(
-                             PathAndInSize({"tests/modules/lenet.jit.pt",
-                                            {{1,1,28,28}}}),
-                             PathAndInSize({"tests/modules/resnet18.jit.pt",
-                                             {{1,3,224,224}}}),
-                             PathAndInSize({"tests/modules/resnet50.jit.pt",
-                                            {{1,3,224,224}}}),
-                             PathAndInSize({"tests/modules/mobilenet_v2.jit.pt",
-                                            {{1,3,224,224}}})));
+                            PathAndInSize({"tests/modules/resnet18.jit.pt",
+                                          {{1,3,224,224}}}),
+                            PathAndInSize({"tests/modules/resnet50.jit.pt",
+                                          {{1,3,224,224}}}),
+                            PathAndInSize({"tests/modules/mobilenet_v2.jit.pt",
+                                          {{1,3,224,224}}})));

tests/modules/test_modules_as_engines.cpp

@@ -19,11 +19,9 @@ TEST_P(ModuleTests, ModuleAsEngineIsClose) {
 INSTANTIATE_TEST_SUITE_P(ModuleAsEngineForwardIsCloseSuite,
                          ModuleTests,
                          testing::Values(
-                             PathAndInSize({"tests/modules/lenet.jit.pt",
-                                            {{1,1,28,28}}}),
-                             PathAndInSize({"tests/modules/resnet18.jit.pt",
-                                            {{1,3,224,224}}}),
-                             PathAndInSize({"tests/modules/resnet50.jit.pt",
-                                            {{1,3,224,224}}}),
-                             PathAndInSize({"tests/modules/mobilenet_v2.jit.pt",
-                                            {{1,3,224,224}}})));
+                            PathAndInSize({"tests/modules/resnet18.jit.pt",
+                                          {{1,3,224,224}}}),
+                            PathAndInSize({"tests/modules/resnet50.jit.pt",
+                                          {{1,3,224,224}}}),
+                            PathAndInSize({"tests/modules/mobilenet_v2.jit.pt",
+                                          {{1,3,224,224}}})));