putting file restore in the finally block

Author: aseembits93

tests/test_instrument_tests.py

@@ -516,12 +516,12 @@ def test_sort():
             testing_time=0.1,
             line_profiler_output_file = line_profiler_output_file
         )
-        func_optimizer.write_code_and_helpers(
-            func_optimizer.function_to_optimize_source_code, original_helper_code, func_optimizer.function_to_optimize.file_path
-        )
         tmp_lpr = list(line_profile_results['timings'].keys())
         assert len(tmp_lpr) == 1 and line_profile_results['timings'][tmp_lpr[0]][0][1]==2
     finally:
+        func_optimizer.write_code_and_helpers(
+            func_optimizer.function_to_optimize_source_code, original_helper_code, func_optimizer.function_to_optimize.file_path
+        )
         test_path.unlink(missing_ok=True)
         test_path_perf.unlink(missing_ok=True)
 
@@ -766,12 +766,12 @@ def test_sort_parametrized(input, expected_output):
             testing_time=0.1,
             line_profiler_output_file = line_profiler_output_file
         )
-        func_optimizer.write_code_and_helpers(
-            func_optimizer.function_to_optimize_source_code, original_helper_code, func_optimizer.function_to_optimize.file_path
-        )
         tmp_lpr = list(line_profile_results['timings'].keys())
         assert len(tmp_lpr) == 1 and line_profile_results['timings'][tmp_lpr[0]][0][1]==3
     finally:
+        func_optimizer.write_code_and_helpers(
+            func_optimizer.function_to_optimize_source_code, original_helper_code, func_optimizer.function_to_optimize.file_path
+        )
         test_path.unlink(missing_ok=True)
         test_path_perf.unlink(missing_ok=True)
 
@@ -1102,12 +1102,12 @@ def test_sort_parametrized_loop(input, expected_output):
             testing_time=0.1,
             line_profiler_output_file = line_profiler_output_file
         )
-        func_optimizer.write_code_and_helpers(
-            func_optimizer.function_to_optimize_source_code, original_helper_code, func_optimizer.function_to_optimize.file_path
-        )
         tmp_lpr = list(line_profile_results['timings'].keys())
         assert len(tmp_lpr) == 1 and line_profile_results['timings'][tmp_lpr[0]][0][1]==6
     finally:
+        func_optimizer.write_code_and_helpers(
+            func_optimizer.function_to_optimize_source_code, original_helper_code, func_optimizer.function_to_optimize.file_path
+        )
         test_path.unlink(missing_ok=True)
         test_path_behavior.unlink(missing_ok=True)
         test_path_perf.unlink(missing_ok=True)
@@ -1372,12 +1372,12 @@ def test_sort():
             testing_time=0.1,
             line_profiler_output_file = line_profiler_output_file
         )
-        func_optimizer.write_code_and_helpers(
-            func_optimizer.function_to_optimize_source_code, original_helper_code, func_optimizer.function_to_optimize.file_path
-        )
         tmp_lpr = list(line_profile_results['timings'].keys())
         assert len(tmp_lpr) == 1 and line_profile_results['timings'][tmp_lpr[0]][0][1]==3
     finally:
+        func_optimizer.write_code_and_helpers(
+            func_optimizer.function_to_optimize_source_code, original_helper_code, func_optimizer.function_to_optimize.file_path
+        )
         test_path.unlink(missing_ok=True)
         test_path_perf.unlink(missing_ok=True)
         test_path_behavior.unlink(missing_ok=True)