fix fps parser

fps/parser.py

@@ -5,6 +5,7 @@ import random
 import string
 import hashlib
 import json
+import os
 import xml.etree.ElementTree as ET
 
 
@@ -145,14 +146,14 @@ class FPSHelper(object):
             if spj:
                 one_info = {
                     "input_size": len(input_content),
-                    "input_name": f"{index}.in"
+                    "input_name": f"{index + 1}.in"
                 }
             else:
                 one_info = {
                     "input_size": len(input_content),
-                    "input_name": f"{index}.in",
+                    "input_name": f"{index + 1}.in",
                     "output_size": len(output_content),
-                    "output_name": f"{index}.out",
+                    "output_name": f"{index + 1}.out",
                     "stripped_output_md5": hashlib.md5(output_content.rstrip().encode("utf-8")).hexdigest()
                 }
             test_cases[index] = one_info
@@ -162,11 +163,11 @@ class FPSHelper(object):
         }
         with open(os.path.join(base_dir, "info"), "w", encoding="utf-8") as f:
             f.write(json.dumps(info, indent=4))
+        return info
 
 
 if __name__ == "__main__":
     import pprint
-    import os
 
     parser = FPSParser("fps.xml")
     helper = FPSHelper()