PY-63319 Fix inverting comparisons in conditions when the left operand doesn't belong to a chained comparison

GitOrigin-RevId: 75b988b6c53bd15d0001ede1f2ebc71153b1de5b

python/python-psi-impl/src/com/jetbrains/python/codeInsight/ConditionUtil.kt

@@ -207,7 +207,7 @@ private fun getInvertedConditionExpression(
     val chainedExpressions = mutableListOf<PyExpression>()
     var currentExpression = expression
     while (currentExpression is PyBinaryExpression && comparisonStrings.containsKey(currentExpression.operator)) {
-      val leftExpression = (currentExpression.leftExpression as? PyBinaryExpression)?.rightExpression
+      val leftExpression = (currentExpression.leftExpression as? PyBinaryExpression)?.takeIf { comparisonStrings.containsKey(it.operator) }?.rightExpression
                            ?: currentExpression.leftExpression
       val invertedOperator = invertedComparisons.getValue(currentExpression.operator)
       val invertedExpression = generator.createBinaryExpression(

python/testData/intentions/PyInvertIfConditionIntentionTest/nonChainableBinaryExpressionAsComparisonLeftOperand.py

@@ -0,0 +1,3 @@
+def func():
+    <caret>if 4 - 3 == 1:
+        print("hi")

python/testData/intentions/PyInvertIfConditionIntentionTest/nonChainableBinaryExpressionAsComparisonLeftOperand_after.py

@@ -0,0 +1,4 @@
+def func():
+    if 4 - 3 != 1:
+        return
+    print("hi")

python/testSrc/com/jetbrains/python/intentions/PyInvertIfConditionIntentionTest.kt

@@ -495,6 +495,11 @@ class PyInvertIfConditionIntentionTest : PyIntentionTestCase() {
     doTest()
   }
 
+  // PY-63319
+  fun testNonChainableBinaryExpressionAsComparisonLeftOperand() {
+    doTest()
+  }
+
   private fun doTest() {
     doIntentionTest(PyPsiBundle.message("INTN.invert.if.condition"))
   }