d

Author: cheqianh

amazon/ionbenchmark/ion_benchmark_cli.py

@@ -344,17 +344,18 @@ def reset_for_each_execution(each_option):
     return api, format_option
 
 
+def generate_ion_file_with_format(file, binary):
+    cf = open("converted_file", "wb")
+    with open(file, "br") as fp:
+        ion.dump(ion.load(fp, single_value=False), cf, binary=binary)
+    return cf.name
+
+
 def convert_file_to_desired_format(f, file):
     if f == Format.ION_BINARY.value:
-        cf = open("converted_file", "wb")
-        with open(file, "br") as fp:
-            ion.dump(ion.load(fp, single_value=False), cf, binary=True)
-        return cf.name
+        return generate_ion_file_with_format(file, True)
     elif f == Format.ION_TEXT.value:
-        cf = open("converted_file", "wb")
-        with open(file, "br") as fp:
-            ion.dump(ion.load(fp, single_value=False), cf, binary=False)
-        return cf.name
+        return generate_ion_file_with_format(file, False)
     else:
         raise Exception('Failed to convert file due to unknown format.')