handle formatting files with no formatting issues

Author: mohammedahmed18

code_to_optimize/no_formatting_errors.py

@@ -0,0 +1,71 @@
+import os, sys, json, datetime, math, random
+import requests
+from collections import defaultdict, OrderedDict
+from typing import List, Dict, Optional, Union, Tuple, Any
+import numpy as np
+import pandas as pd
+
+# This is a poorly formatted Python file with many style violations
+
+
+class UnformattedExampleClass(object):
+    def __init__(
+        self,
+        name,
+        age=None,
+        email=None,
+        phone=None,
+        address=None,
+        city=None,
+        state=None,
+        zip_code=None,
+    ):
+        self.name = name
+        self.age = age
+        self.email = email
+        self.phone = phone
+        self.address = address
+        self.city = city
+        self.state = state
+        self.zip_code = zip_code
+        self.data = {"name": name, "age": age, "email": email}
+
+    def get_info(self):
+        return f"Name: {self.name}, Age: {self.age}"
+
+    def update_data(self, **kwargs):
+        for key, value in kwargs.items():
+            if hasattr(self, key):
+                setattr(self, key, value)
+        self.data.update(kwargs)
+
+
+def process_data(
+    data_list, filter_func=None, transform_func=None, sort_key=None, reverse=False
+):
+    if not data_list:
+        return []
+    if filter_func:
+        data_list = [item for item in data_list if filter_func(item)]
+    if transform_func:
+        data_list = [transform_func(item) for item in data_list]
+    if sort_key:
+        data_list = sorted(data_list, key=sort_key, reverse=reverse)
+    return data_list
+
+
+def calculate_statistics(numbers):
+    if not numbers:
+        return None
+    mean = sum(numbers) / len(numbers)
+    median = sorted(numbers)[len(numbers) // 2]
+    variance = sum((x - mean) ** 2 for x in numbers) / len(numbers)
+    std_dev = math.sqrt(variance)
+    return {
+        "mean": mean,
+        "median": median,
+        "variance": variance,
+        "std_dev": std_dev,
+        "min": min(numbers),
+        "max": max(numbers),
+    }

codeflash/code_utils/formatter.py

@@ -15,11 +15,11 @@
 
 def get_diff_output_by_black(filepath: str, unformatted_content: str) -> Optional[str]:
     try:
-        from black import Mode, format_file_contents, output
+        from black import Mode, format_file_contents, output, report
 
         formatted_content = format_file_contents(src_contents=unformatted_content, fast=True, mode=Mode())
         return output.diff(unformatted_content, formatted_content, a_name=filepath, b_name=filepath)
-    except ImportError:
+    except (ImportError, report.NothingChanged):
         return None
 
 

tests/test_formatter.py

@@ -279,3 +279,7 @@ def test_formatting_file_with_many_diffs():
 def test_formatting_file_with_few_diffs():
     """Test that files with few formatting errors are formatted (content changed)."""
     _run_formatting_test("few_formatting_errors.py", should_content_change=True)
+
+def test_formatting_file_with_no_diffs():
+    """Test that files with no formatting errors are unchanged."""
+    _run_formatting_test("no_formatting_errors.py", should_content_change=False)