Artemis: Refactor sum_range and sum_square Functions for Efficiency and Readability

src/llm_benchmark/control/double.py

@@ -4,20 +4,7 @@
 class DoubleForLoop:
     @staticmethod
     def sum_square(n: int) -> int:
-        """Sum of squares of numbers from 0 to n (exclusive)
-
-        Args:
-            n (int): Number to sum up to
-
-        Returns:
-            int: Sum of squares of numbers from 0 to n
-        """
-        sum_ = 0
-        for i in range(n):
-            for j in range(n):
-                if i == j:
-                    sum_ += i * j
-        return sum_
+        return sum(i * i for i in range(n))
 
     @staticmethod
     def sum_triangle(n: int) -> int:
@@ -90,4 +77,4 @@ def sum_matrix(m: List[List[int]]) -> int:
         for i in range(len(m)):
             for j in range(len(m[i])):
                 sum_ += m[i][j]
-        return sum_
+        return sum_

src/llm_benchmark/control/single.py

@@ -5,17 +5,14 @@ class SingleForLoop:
     @staticmethod
     def sum_range(n: int) -> int:
         """Sum of range of numbers from 0 to n
-
+        
         Args:
             n (int): Number to sum up to
-
+        
         Returns:
             int: Sum of range of numbers from 0 to n
         """
-        arr = []
-        for i in range(n):
-            arr.append(i)
-        return sum(arr)
+        return (n * (n + 1)) // 2
 
     @staticmethod
     def max_list(v: List[int]) -> int:
@@ -48,4 +45,4 @@ def sum_modulus(n: int, m: int) -> int:
         for i in range(n):
             if i % m == 0:
                 arr.append(i)
-        return sum(arr)
+        return sum(arr)