Changed method to mutate a line of code

servo · Nov 6, 2017 · 6688b8a · 6688b8a
1 parent c2d46c3
commit 6688b8a
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/python/servo/mutation/test.py b/python/servo/mutation/test.py
@@ -14,11 +14,10 @@
 
 
 def mutate_line(file_name, line_number):
-    lines = open(file_name, 'r').readlines()
-    lines[line_number - 1] = re.sub(r'\s&&\s', ' || ', lines[line_number - 1])
-    out = open(file_name, 'w')
-    out.writelines(lines)
-    out.close()
+    for line in fileinput.input(file_name, inplace=True):
+        if(fileinput.lineno() == line_number):
+            line = re.sub(r'\s&&\s', ' || ', line)
+        print line.rstrip()
 
 
 def mutation_test(file_name, tests):