Handle typo in float call more gracefully

Author: Pierre-Sassoulas

pylint/extensions/code_style.py

@@ -4,6 +4,7 @@
 
 from __future__ import annotations
 
+import difflib
 from typing import TYPE_CHECKING, TypeGuard, cast
 
 from astroid import nodes
@@ -127,14 +128,21 @@ def visit_call(self, node: nodes.Call) -> None:
                         c.isalpha() and c.lower() != "e" for c in node.args[0].value
                     )
                 ):
-                    is_nan = "nan" in node.args[0].value.lower()
-                    minus = (
-                        "-" if not is_nan and node.args[0].value.startswith("-") else ""
-                    )
+                    value = node.args[0].value.lower()
+                    math_call: str
+                    if "nan" in value:
+                        math_call = "nan"
+                    elif "inf" in value:
+                        math_call = "inf"
+                    else:
+                        math_call = difflib.get_close_matches(
+                            value, ["inf", "nan"], n=1, cutoff=0
+                        )[0]
+                    minus = "-" if math_call == "inf" and value.startswith("-") else ""
                     self.add_message(
                         "use-math-not-float",
                         node=node,
-                        args=(minus, "nan" if is_nan else "inf", node.as_string()),
+                        args=(minus, math_call, node.as_string()),
                         confidence=INFERENCE,
                     )
 

tests/functional/ext/code_style/cs_use_math_not_float.py

@@ -14,3 +14,5 @@
 neg_nan_float = float("-nan")  # [use-math-not-float]
 pos_nan_float = float("+nan")  # [use-math-not-float]
 upper_nan_float = float("NaN")  # [use-math-not-float]
+typo_nan_float = float("nani")  # [use-math-not-float]
+other_typo_nan_float = float("nna")  # [use-math-not-float]

tests/functional/ext/code_style/cs_use_math_not_float.txt

@@ -9,3 +9,5 @@ use-math-not-float:13:12:13:24::math.nan is preferable to float('nan'):INFERENCE
 use-math-not-float:14:16:14:29::math.nan is preferable to float('-nan'):INFERENCE
 use-math-not-float:15:16:15:29::math.nan is preferable to float('+nan'):INFERENCE
 use-math-not-float:16:18:16:30::math.nan is preferable to float('NaN'):INFERENCE
+use-math-not-float:17:17:17:30::math.nan is preferable to float('nani'):INFERENCE
+use-math-not-float:18:23:18:35::math.nan is preferable to float('nna'):INFERENCE