PY-37720 Impl python ml completion features

GitOrigin-RevId: 3b8029ab460672b286d7b1cf96fd35bf9c57bf59

python/testData/codeInsight/mlcompletion/classBuiltins.py

@@ -0,0 +1 @@
+a = E<caret>

python/testData/codeInsight/mlcompletion/classNotBuiltins.py

@@ -0,0 +1,4 @@
+class MyClazz:
+  pass
+
+a = M<caret>

python/testData/codeInsight/mlcompletion/functionBuiltins.py

@@ -0,0 +1 @@
+ma<caret>

python/testData/codeInsight/mlcompletion/functionNotBuiltins.py

@@ -0,0 +1,5 @@
+def my_not_builtins_function(x, y):
+  print(x)
+  print(y)
+
+m<caret>

python/testData/codeInsight/mlcompletion/isAfterIfWithoutElse1.py

@@ -0,0 +1,6 @@
+a = input()
+b = input()
+something = a > b
+if something:
+  print(a)
+<caret>

python/testData/codeInsight/mlcompletion/isAfterIfWithoutElse2.py

@@ -0,0 +1,8 @@
+a = input()
+b = input()
+something = a > b
+if something:
+  print(a)
+else:
+  print(b)
+<caret>

python/testData/codeInsight/mlcompletion/isAfterIfWithoutElse3.py

@@ -0,0 +1,7 @@
+a = input()
+b = input()
+something = a > b
+if something:
+  print(a)
+c = input()
+<caret>

python/testData/codeInsight/mlcompletion/isAfterIfWithoutElse4.py

@@ -0,0 +1,8 @@
+a = input()
+b = input()
+something = a > b
+if something:
+  print(a)
+elif something:
+  print(b)
+e<caret>

python/testData/codeInsight/mlcompletion/isAfterIfWithoutElse5.py

@@ -0,0 +1,6 @@
+if True:
+    if True:
+        pass
+    else:
+        pass
+    <caret>

python/testData/codeInsight/mlcompletion/isDirectlyInArgumentsContext1.py

@@ -0,0 +1 @@
+print(<caret>)

python/testData/codeInsight/mlcompletion/isDirectlyInArgumentsContext2.py

@@ -0,0 +1,3 @@
+some_variable1 = 1
+some_variable2 = 2
+print(some_variable1, some_<caret>)

python/testData/codeInsight/mlcompletion/isDirectlyInArgumentsContext3.py

@@ -0,0 +1,3 @@
+some_variable1 = 1
+some_variable2 = 2
+print(some_variable1, end=<caret>)

python/testData/codeInsight/mlcompletion/isDirectlyInArgumentsContext4.py

@@ -0,0 +1,7 @@
+some_variable1 = 1
+some_variable2 = 2
+
+def foo(some_param):
+  print(some_param)
+
+print(some_variable1, foo(<caret>))

python/testData/codeInsight/mlcompletion/isInCondition1.py

@@ -0,0 +1 @@
+if <caret>

python/testData/codeInsight/mlcompletion/isInCondition2.py

@@ -0,0 +1,5 @@
+var1 = bool(input())
+var2 = bool(input())
+if var1:
+  print(var2)
+elif v<caret>

python/testData/codeInsight/mlcompletion/isInCondition3.py

@@ -0,0 +1,3 @@
+var1 = bool(input())
+if var1:
+  <caret>

python/testData/codeInsight/mlcompletion/isInCondition4.py

@@ -0,0 +1,4 @@
+def foo(x):
+  return x == 3
+
+if foo(<caret>)

python/testData/codeInsight/mlcompletion/isInCondition5.py

@@ -0,0 +1,3 @@
+var1 = bool(input())
+if var1:
+  p<caret>

python/testData/codeInsight/mlcompletion/isInCondition6.py

@@ -0,0 +1,4 @@
+arr1 = [1, 2, 3]
+arr2 = [4, 5, 6]
+b = 3
+while b in <caret>:

python/testData/codeInsight/mlcompletion/isInFor1.py

@@ -0,0 +1 @@
+for <caret>

python/testData/codeInsight/mlcompletion/isInFor2.py

@@ -0,0 +1,2 @@
+my_list = [1, 2, 3]
+for e<caret> in my_list:

python/testData/codeInsight/mlcompletion/isInFor3.py

@@ -0,0 +1,2 @@
+my_list = [1, 2, 3]
+for e in <caret>:

python/testData/codeInsight/mlcompletion/isInFor4.py

@@ -0,0 +1,2 @@
+my_list = [1, 2, 3]
+for e in m<caret>:

python/testData/codeInsight/mlcompletion/isInFor5.py

@@ -0,0 +1,3 @@
+my_list = [1, 2, 3]
+for e in my_list:
+  <caret>

python/testData/codeInsight/mlcompletion/kindFromTarget1.py

@@ -0,0 +1,2 @@
+local_variable = 42
+print(lo<caret>)

python/testData/codeInsight/mlcompletion/kindFromTarget2.py

@@ -0,0 +1,6 @@
+def foo(f):
+  for line in f:
+    print(line)
+
+with open("file", "r") as as_target:
+  foo(<caret>)

python/testData/codeInsight/mlcompletion/kindKeyword.py

@@ -0,0 +1 @@
+i<caret>

python/testData/codeInsight/mlcompletion/kindNamedArg.py

@@ -0,0 +1 @@
+print("test", se<caret>)

python/testData/codeInsight/mlcompletion/kindPackageOrModule.py

@@ -0,0 +1 @@
+import s<caret>

python/testData/codeInsight/mlcompletion/numberOfOccurrencesClass.py

@@ -0,0 +1,8 @@
+class MyClazz:
+  pass
+
+a = MyClazz()
+
+def foo(x):
+  b = MyClazz()
+  c = M<caret>

python/testData/codeInsight/mlcompletion/numberOfOccurrencesFunction.py

@@ -0,0 +1,2 @@
+a = min(1, 2)
+b = m<caret>

python/testData/codeInsight/mlcompletion/numberOfOccurrencesNamedArgs1.py

@@ -0,0 +1 @@
+print("a", end=".", en<caret>)

python/testData/codeInsight/mlcompletion/numberOfOccurrencesNamedArgs2.py

@@ -0,0 +1 @@
+print("a", end=".", <caret>)

python/testData/codeInsight/mlcompletion/numberOfOccurrencesPackagesOrModules.py

@@ -0,0 +1,2 @@
+import collections
+import o<caret>

python/testData/codeInsight/mlcompletion/prevNeighbourKeywords1.py

@@ -0,0 +1,3 @@
+arr1 = [1, 2, 3]
+arr2 = [4, 5, 6]
+if a in <caret>

python/testData/codeInsight/mlcompletion/prevNeighbourKeywords2.py

@@ -0,0 +1,3 @@
+arr1 = [1, 2, 3]
+arr2 = [4, 5, 6]
+if a not in ar<caret>

python/testData/codeInsight/mlcompletion/sameColumnKeywords1.py

@@ -0,0 +1,6 @@
+a, b = 2, 3
+if a > b:
+  print(a)
+elif a * 2 > b:
+  print(b)
+<caret>

python/testData/codeInsight/mlcompletion/sameColumnKeywords2.py

@@ -0,0 +1,7 @@
+a, b = 2, 3
+if a > b:
+  print(a)
+z = 42
+if a * 2 > b:
+  print(b)
+<caret>

python/testData/codeInsight/mlcompletion/sameColumnKeywords3.py

@@ -0,0 +1,10 @@
+
+
+def aaa():
+    pass
+
+
+def bbb():
+    pass
+
+<caret>

python/testData/codeInsight/mlcompletion/sameColumnKeywords4.py

@@ -0,0 +1,12 @@
+
+class Clz:
+  def aaa(self):
+      for i in range(0, 10):
+        pass
+
+
+  def bbb(self):
+      for i in range(0, 10):
+        pass
+
+  <caret>

python/testData/codeInsight/mlcompletion/sameColumnKeywords5.py

@@ -0,0 +1,13 @@
+a, b = 2, 3
+if a > b:
+
+  print(a)
+
+  for i in range(0, 10)
+    pass
+
+
+  if a * 2 > b:
+    print(b)
+
+  <caret>

python/testData/codeInsight/mlcompletion/sameColumnKeywords6.py

@@ -0,0 +1,14 @@
+a, b = 2, 3
+if a > b:
+
+  print(a)
+
+  for i in range(0, 10)
+    pass
+
+  z = 42
+
+  if a * 2 > b:
+    print(b)
+
+  <caret>

python/testData/codeInsight/mlcompletion/sameLineKeywords1.py

@@ -0,0 +1 @@
+if a in <caret>

python/testData/codeInsight/mlcompletion/sameLineKeywords2.py

@@ -0,0 +1,2 @@
+if not in lambda
+if a in <caret>