not x is bool for any x (#32)

orsinium · web-flow · commit d309e44e9fdb · 2022-07-03T07:14:24.000-07:00
diff --git a/autotyping/autotyping.py b/autotyping/autotyping.py
@@ -540,6 +540,10 @@ def type_of_expression(expr: libcst.BaseExpression) -> Optional[Type[object]]:
             return None
     elif isinstance(expr, libcst.Name) and expr.value in ("True", "False"):
         return bool
+    elif isinstance(expr, libcst.UnaryOperation) and isinstance(
+        expr.operator, libcst.Not
+    ):
+        return bool
     elif (
         isinstance(expr, libcst.Call)
         and isinstance(expr.func, libcst.Attribute)
diff --git a/tests/test_codemod.py b/tests/test_codemod.py
@@ -49,6 +49,9 @@ def foo():
             def bar():
                 return 1
 
+            def return_not(x):
+                return not x
+
             def formatter(x):
                 return "{}".format(x)
 
@@ -68,6 +71,9 @@ def foo():
             def bar() -> int:
                 return 1
 
+            def return_not(x) -> bool:
+                return not x
+
             def formatter(x) -> str:
                 return "{}".format(x)
 
