update pytest saveload case, test=model

framework/e2e/research/saveload_ten_model/model1.py

@@ -48,7 +48,6 @@ def jit_save(model, path, inputs):
 
     paddle.jit.save(model, path)
 
-
     return pred, outputs_mean
 
 
@@ -164,8 +163,8 @@ def forward(self, x):
         return x
 
 
-if __name__ == "__main__":
-
+def test_model1():
+    """test_model1"""
     # 转换为Paddle张量
     num_image = 10
     size = 64
@@ -175,9 +174,9 @@ def forward(self, x):
     # assert (size % (2 ** max_conv_layers)) != 0, "不应该能够整除，但却整除了！"
 
     data = np.random.randn(num_image, 3, size, size).astype("float32")
-    label = np.random.randint(0, 10, (10, 1), dtype="int64")
+    # label = np.random.randint(0, 10, (10, 1), dtype="int64")
     inputs = paddle.to_tensor(data)
-    labels = paddle.to_tensor(label)
+    # labels = paddle.to_tensor(label)
     path = "simple/model1/demo1"  # 路径这里用demo，若改infer中对应需要改
 
     model = RandomNet()