Merge branch 'main' into GCI103-python

Author: dedece35

CHANGELOG.md

@@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 
 - [#76](https://github.com/green-code-initiative/creedengo-python/pull/76) Add rule GCI 103 Dictionary Items Unused. A rule specifying that dictionary iteration should consider the pertinence of the element used.
+- [#79](https://github.com/green-code-initiative/creedengo-python/pull/79) Add rule GCI106 Avoid SQRT in a loop
 - [#71](https://github.com/green-code-initiative/creedengo-python/pull/71) Add rule GCI96 Require Usecols Argument in Pandas Read Functions
 - [#72](https://github.com/green-code-initiative/creedengo-python/pull/72) Add rule GCI97 Optimize square computation (scalar vs vectorized method)
 

src/it/java/org/greencodeinitiative/creedengo/python/integration/tests/GCIRulesIT.java

@@ -320,4 +320,18 @@ void testGCI103(){
 
     }
 
+    @Test
+    void testGCI106() {
+        String filePath = "src/avoidSqrtInLoop.py";
+        String ruleId = "creedengo-python:GCI106";
+        String ruleMsg = "Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.";
+        int[] startLines = new int[]{
+            7, 11, 16, 21, 45, 52, 60
+        };
+        int[] endLines = new int[]{
+            7, 11, 16, 21, 45, 52, 60
+        };
+        checkIssuesForFile(filePath, ruleId, ruleMsg, startLines, endLines, SEVERITY, TYPE, EFFORT_5MIN);
+    }
+
 }

src/it/test-projects/creedengo-python-plugin-test-project/src/avoidSqrtInLoop.py

@@ -0,0 +1,75 @@
+import math
+import numpy as np
+
+results = []
+numbers = [1, 2, 3, 4, 5]
+for n in numbers:
+    results.append(math.sqrt(n))  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+
+results = []
+for x in [9, 16, 25]:
+    results.append(np.sqrt(x))  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+
+results = []
+i = 0
+while i < 3:
+    results.append(np.sqrt(i))  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+    i += 1
+
+results = []
+for n in range(10):
+    val = math.sqrt(n)  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+    results.append(val)
+
+class DataProcessor:
+    numbers = np.array([1, 4, 9])
+    sqrt_values = np.sqrt(numbers)
+
+x = math.sqrt(25)
+
+class Analyzer:
+    def __init__(self, data):
+        self.results = np.sqrt(np.array(data))
+
+intermediate = []
+for n in range(3):
+    intermediate.append(n)
+final = np.sqrt(np.array(intermediate))
+
+results = np.sqrt([x for x in range(5)])
+
+def non_compliant_math_sqrt_in_for_loop():
+    numbers = [1, 2, 3, 4, 5]
+    results = []
+    for num in numbers:
+        results.append(math.sqrt(num))  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+    return results
+
+def non_compliant_numpy_scalar_sqrt_in_for_loop():
+    numbers = [1, 2, 3, 4, 5]
+    results = []
+    for num in numbers:
+        results.append(np.sqrt(num))  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+    return results
+
+def non_compliant_math_sqrt_in_while_loop():
+    numbers = [1, 2, 3, 4, 5]
+    results = []
+    i = 0
+    while i < len(numbers):
+        results.append(math.sqrt(numbers[i]))  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+        i += 1
+    return results
+
+def compliant_numpy_vectorized_sqrt():
+    numbers = np.array([1, 2, 3, 4, 5])
+    results = np.sqrt(numbers)
+    return results
+
+def compliant_list_comprehension_with_vectorized_operation():
+    numbers = [1, 2, 3, 4, 5]
+    return np.sqrt(np.array(numbers))
+
+def compliant_math_sqrt_outside_loop():
+    value = 16
+    return math.sqrt(value)

src/main/java/org/greencodeinitiative/creedengo/python/PythonRuleRepository.java

@@ -43,6 +43,7 @@ public class PythonRuleRepository implements RulesDefinition, PythonCustomRuleRe
             AvoidMultipleIfElseStatementCheck.class,
             PandasRequireUsecolsArgument.class,
             OptimizeSquareComputation.class,
+            AvoidSqrtInLoop.class,
             DictionaryItemsUnused.class
     );
 

src/main/java/org/greencodeinitiative/creedengo/python/checks/AvoidSqrtInLoop.java

@@ -0,0 +1,75 @@
+/*
+ * creedengo - Python language - Provides rules to reduce the environmental footprint of your Python programs
+ * Copyright © 2024 Green Code Initiative (https://green-code-initiative.org)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.greencodeinitiative.creedengo.python.checks;
+
+import org.sonar.check.Rule;
+import org.sonar.plugins.python.api.PythonSubscriptionCheck;
+import org.sonar.plugins.python.api.SubscriptionContext;
+import org.sonar.plugins.python.api.tree.CallExpression;
+import org.sonar.plugins.python.api.tree.Expression;
+import org.sonar.plugins.python.api.tree.QualifiedExpression;
+import org.sonar.plugins.python.api.tree.Tree;
+
+@Rule(key = "GCI106")
+public class AvoidSqrtInLoop extends PythonSubscriptionCheck {
+
+    public static final String DESCRIPTION = "Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.";
+
+    @Override
+    public void initialize(Context context) {
+        context.registerSyntaxNodeConsumer(Tree.Kind.CALL_EXPR, this::checkCallExpression);
+    }
+
+    private void checkCallExpression(SubscriptionContext context) {
+        CallExpression callExpression = (CallExpression) context.syntaxNode();
+        
+        if (isSqrtCall(callExpression) && hasLoopParent(callExpression)) {
+            context.addIssue(callExpression, DESCRIPTION);
+        }
+    }
+    
+    private boolean isSqrtCall(CallExpression callExpression) {
+        Expression callee = callExpression.callee();
+        
+        // Check for direct calls to math.sqrt
+        if (callee.is(Tree.Kind.QUALIFIED_EXPR)) {
+            QualifiedExpression qualifiedExpression = (QualifiedExpression) callee;
+            String methodName = qualifiedExpression.name().name();
+            
+            if ("sqrt".equals(methodName)) {
+                Expression qualifier = qualifiedExpression.qualifier();
+                if (qualifier != null && qualifier.is(Tree.Kind.NAME)) {
+                    String qualifierName = qualifier.firstToken().value();
+                    return "math".equals(qualifierName) || "np".equals(qualifierName) || "numpy".equals(qualifierName);
+                }
+            }
+        }
+        
+        return false;
+    }
+    
+    private boolean hasLoopParent(Tree tree) {
+        for (Tree parent = tree.parent(); parent != null; parent = parent.parent()) {
+            Tree.Kind kind = parent.getKind();
+            if (kind == Tree.Kind.FOR_STMT || kind == Tree.Kind.WHILE_STMT) {
+                return true;
+            }
+        }
+        return false;
+    }
+}

src/main/resources/org/greencodeinitiative/creedengo/python/creedengo_way_profile.json

@@ -13,6 +13,7 @@
 		"GCI96",
 		"GCI97",
 		"GCI103",
+		"GCI106",
 		"GCI203",
 		"GCI404"
 	]

src/test/java/org/greencodeinitiative/creedengo/python/checks/AvoidSqrtInLoopTest.java

@@ -0,0 +1,28 @@
+/*
+ * creedengo - Python language - Provides rules to reduce the environmental footprint of your Python programs
+ * Copyright © 2024 Green Code Initiative (https://green-code-initiative.org)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.greencodeinitiative.creedengo.python.checks;
+
+import org.junit.Test;
+import org.sonar.python.checks.utils.PythonCheckVerifier;
+
+public class AvoidSqrtInLoopTest {
+    @Test
+    public void test() {
+        PythonCheckVerifier.verify("src/test/resources/checks/avoidSqrtInLoop.py", new AvoidSqrtInLoop());
+    }
+}

src/test/resources/checks/avoidSqrtInLoop.py

@@ -0,0 +1,75 @@
+import math
+import numpy as np
+
+results = []
+numbers = [1, 2, 3, 4, 5]
+for n in numbers:
+    results.append(math.sqrt(n))  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+
+results = []
+for x in [9, 16, 25]:
+    results.append(np.sqrt(x))  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+
+results = []
+i = 0
+while i < 3:
+    results.append(np.sqrt(i))  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+    i += 1
+
+results = []
+for n in range(10):
+    val = math.sqrt(n)  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+    results.append(val)
+
+class DataProcessor:
+    numbers = np.array([1, 4, 9])
+    sqrt_values = np.sqrt(numbers)
+
+x = math.sqrt(25)
+
+class Analyzer:
+    def __init__(self, data):
+        self.results = np.sqrt(np.array(data))
+
+intermediate = []
+for n in range(3):
+    intermediate.append(n)
+final = np.sqrt(np.array(intermediate))
+
+results = np.sqrt([x for x in range(5)])
+
+def non_compliant_math_sqrt_in_for_loop():
+    numbers = [1, 2, 3, 4, 5]
+    results = []
+    for num in numbers:
+        results.append(math.sqrt(num))  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+    return results
+
+def non_compliant_numpy_scalar_sqrt_in_for_loop():
+    numbers = [1, 2, 3, 4, 5]
+    results = []
+    for num in numbers:
+        results.append(np.sqrt(num))  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+    return results
+
+def non_compliant_math_sqrt_in_while_loop():
+    numbers = [1, 2, 3, 4, 5]
+    results = []
+    i = 0
+    while i < len(numbers):
+        results.append(math.sqrt(numbers[i]))  # Noncompliant {{Avoid using scalar sqrt functions in loops. Apply vectorized sqrt operations on arrays directly.}}
+        i += 1
+    return results
+
+def compliant_numpy_vectorized_sqrt():
+    numbers = np.array([1, 2, 3, 4, 5])
+    results = np.sqrt(numbers)
+    return results
+
+def compliant_list_comprehension_with_vectorized_operation():
+    numbers = [1, 2, 3, 4, 5]
+    return np.sqrt(np.array(numbers))
+
+def compliant_math_sqrt_outside_loop():
+    value = 16
+    return math.sqrt(value)