@zzzen/pyright-internal 1.1.254 → 1.1.255

package/src/tests/fourslash/completions.call.typedDict.fourslash.ts

@@ -0,0 +1,247 @@
+/// <reference path="fourslash.ts" />
+
+// @filename: test.py
+//// from typing import TypedDict, Optional, Union, List, Dict, Any
+////
+//// class Movie(TypedDict):
+////     name: str
+////     age: int
+////
+//// def thing(movie: Movie):
+////     pass
+////
+//// thing({'[|/*marker1*/|]'})
+//// thing({'name': '[|/*marker2*/|]'})
+//// thing({'name': 'Robert','[|/*marker3*/|]'})
+//// thing({'name': 'Robert',   '[|/*marker4*/|]'})
+//// thing('[|/*marker5*/|]')
+//// thing({'na[|/*marker6*/|]'})
+//// thing({[|/*marker7*/|]})
+//// thing({'a', '[|/*marker8*/|]'})
+////
+//// class Episode(TypedDict):
+////     title: str
+////     score: int
+////
+//// def thing2(item: Union[Episode, Movie]):
+////    pass
+////
+//// thing2({'[|/*marker9*/|]'})
+//// thing2({'unknown': 'a', '[|/*marker10*/|]': ''})
+//// thing2({'title': 'Episode 01', '[|/*marker11*/|]': ''})
+////
+//// class Wrapper(TypedDict):
+////     age: int
+////     wrapped: Union[bool, Movie]
+////     data: Dict[str, Any]
+////
+//// def thing3(wrapper: Optional[Wrapper]):
+////     pass
+////
+//// thing3({'data': {'[|/*marker12*/|]'}})
+//// thing3({'wrapped': {'[|/*marker13*/|]'}})
+//// thing3({'age': 1, 'wrapped': {'[|/*marker14*/|]'}})
+//// thing3({'unknown': {'[|/*marker15*/|]'}})
+//// thing3({'age': {'[|/*marker16*/|]'}})
+//// thing3({'wrapped': {'name': 'ET', '[|/*marker17*/|]'}})
+
+{
+    const marker1Range = helper.expandPositionRange(helper.getPositionRange('marker1'), 1, 1);
+    const marker3Range = helper.expandPositionRange(helper.getPositionRange('marker3'), 1, 1);
+    const marker4Range = helper.expandPositionRange(helper.getPositionRange('marker4'), 1, 1);
+    const marker6Range = helper.expandPositionRange(helper.getPositionRange('marker6'), 3, 1);
+    const marker7Range = helper.getPositionRange('marker7');
+    const marker8Range = helper.expandPositionRange(helper.getPositionRange('marker8'), 1, 1);
+    const marker9Range = helper.expandPositionRange(helper.getPositionRange('marker9'), 1, 1);
+    const marker10Range = helper.expandPositionRange(helper.getPositionRange('marker10'), 1, 1);
+    const marker11Range = helper.expandPositionRange(helper.getPositionRange('marker11'), 1, 1);
+    const marker13Range = helper.expandPositionRange(helper.getPositionRange('marker13'), 1, 1);
+    const marker14Range = helper.expandPositionRange(helper.getPositionRange('marker14'), 1, 1);
+    const marker17Range = helper.expandPositionRange(helper.getPositionRange('marker17'), 1, 1);
+
+    // @ts-ignore
+    await helper.verifyCompletion('exact', 'markdown', {
+        marker1: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker1Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker1Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker2: {
+            completions: [],
+        },
+        marker3: {
+            completions: [
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker3Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker4: {
+            completions: [
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker4Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker5: {
+            completions: [],
+        },
+        marker6: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker6Range, newText: "'name'" },
+                },
+            ],
+        },
+        marker8: {
+            completions: [
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker8Range, newText: "'age'" },
+                },
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker8Range, newText: "'name'" },
+                },
+            ],
+        },
+        marker9: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker9Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker9Range, newText: "'age'" },
+                },
+                {
+                    label: "'title'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker9Range, newText: "'title'" },
+                },
+                {
+                    label: "'score'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker9Range, newText: "'score'" },
+                },
+            ],
+        },
+        marker10: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker10Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker10Range, newText: "'age'" },
+                },
+                {
+                    label: "'title'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker10Range, newText: "'title'" },
+                },
+                {
+                    label: "'score'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker10Range, newText: "'score'" },
+                },
+            ],
+        },
+        marker11: {
+            completions: [
+                {
+                    label: "'score'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker11Range, newText: "'score'" },
+                },
+            ],
+        },
+        marker12: {
+            completions: [],
+        },
+        marker13: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker13Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker13Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker14: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker14Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker14Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker15: {
+            completions: [],
+        },
+        marker16: {
+            completions: [],
+        },
+        marker17: {
+            completions: [
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker17Range, newText: "'age'" },
+                },
+            ],
+        },
+    });
+
+    // @ts-ignore
+    await helper.verifyCompletion('included', 'markdown', {
+        marker7: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker7Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker7Range, newText: "'age'" },
+                },
+            ],
+        },
+    });
+}

package/src/tests/fourslash/completions.call.typedDict.list.fourslash.ts

@@ -0,0 +1,163 @@
+/// <reference path="fourslash.ts" />
+
+// @filename: test.py
+//// from typing import TypedDict, Union, List
+////
+//// class Movie(TypedDict):
+////     name: str
+////     age: int
+////
+//// class MultipleInputs(TypedDict):
+////     items: List[Movie]
+////     union: Union[bool, List[Movie]]
+////     unions: Union[Movie, Union[bool, List[Movie]]]
+////
+//// def thing(inputs: MultipleInputs):
+////     pass
+////
+//// thing({'items': ['[|/*marker1*/|]']})
+//// thing({'items': {'[|/*marker2*/|]'}})
+//// thing({'items': [{'[|/*marker3*/|]'}]})
+//// thing({'union': [{'[|/*marker4*/|]'}]})
+//// thing({'unions': {'[|/*marker5*/|]'}})
+//// thing({'unions': [{'[|/*marker6*/|]'}]})
+////
+//// def thing2(movies: List[Movie]):
+////     pass
+////
+//// thing2([{'[|/*marker7*/|]'}])
+//// thing2({'[|/*marker8*/|]'})
+////
+//// class Wrapper(TypedDict):
+////     wrapped: MultipleInputs
+////
+//// def thing3(wrapper: Wrapper):
+////     pass
+////
+//// thing3({'wrapped': {'items': [{'[|/*marker9*/|]'}]}})
+//// thing3({'wrapped': {'items': {'[|/*marker10*/|]'}}})
+//// thing3({'wrapped': {'items': [{'a': 'b'}, {'[|/*marker11*/|]'}]}})
+
+{
+    const marker3Range = helper.expandPositionRange(helper.getPositionRange('marker3'), 1, 1);
+    const marker4Range = helper.expandPositionRange(helper.getPositionRange('marker4'), 1, 1);
+    const marker5Range = helper.expandPositionRange(helper.getPositionRange('marker5'), 1, 1);
+    const marker6Range = helper.expandPositionRange(helper.getPositionRange('marker6'), 1, 1);
+    const marker7Range = helper.expandPositionRange(helper.getPositionRange('marker7'), 1, 1);
+    const marker9Range = helper.expandPositionRange(helper.getPositionRange('marker9'), 1, 1);
+    const marker11Range = helper.expandPositionRange(helper.getPositionRange('marker11'), 1, 1);
+
+    // @ts-ignore
+    await helper.verifyCompletion('exact', 'markdown', {
+        marker1: {
+            completions: [],
+        },
+        marker2: {
+            completions: [],
+        },
+        marker3: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker3Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker3Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker4: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker4Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker4Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker5: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker5Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker5Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker6: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker6Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker6Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker7: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker7Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker7Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker8: {
+            completions: [],
+        },
+        marker9: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker9Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker9Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker10: {
+            completions: [],
+        },
+        marker11: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker11Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker11Range, newText: "'age'" },
+                },
+            ],
+        },
+    });
+}

package/src/tests/fourslash/completions.call.typedDict.states.fourslash.ts

@@ -0,0 +1,135 @@
+/// <reference path="fourslash.ts" />
+
+// @filename: test.py
+//// from typing import TypedDict
+////
+//// class Movie(TypedDict):
+////     name: str
+////     age: int
+////
+//// def thing(movie: Movie):
+////     pass
+////
+//// thing(movie={'foo': 'a', '[|/*marker1*/|]'})
+//// thing(movie={'foo': 'a', 'a[|/*marker2*/|]'})
+//// thing(
+////     movie={
+////        'name': 'Parasite',
+////        '[|/*marker3*/|]
+////     }
+//// )
+//// thing(
+////     movie={
+////        'name': 'Parasite',
+////        '[|/*marker4*/|]'
+////     }
+//// )
+//// thing({
+////     'name': 'Parasite',
+////     # hello world
+////     '[|/*marker5*/|]'
+//// })
+//// thing({'foo': '[|/*marker6*/|]'})
+
+{
+    // completions that rely on token parsing instead of node parsing
+    const marker1Range = helper.expandPositionRange(helper.getPositionRange('marker1'), 1, 1);
+    const marker2Range = helper.expandPositionRange(helper.getPositionRange('marker2'), 2, 1);
+    const marker3Range = helper.expandPositionRange(helper.getPositionRange('marker3'), 1, 0);
+    const marker4Range = helper.expandPositionRange(helper.getPositionRange('marker4'), 1, 1);
+    const marker5Range = helper.expandPositionRange(helper.getPositionRange('marker5'), 1, 1);
+
+    // @ts-ignore
+    await helper.verifyCompletion('exact', 'markdown', {
+        marker1: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker1Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker1Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker2: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker2Range, newText: "'name'" },
+                },
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker2Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker6: {
+            completions: [],
+        },
+    });
+
+    // @ts-ignore
+    await helper.verifyCompletion('included', 'markdown', {
+        marker3: {
+            completions: [
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker3Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker4: {
+            completions: [
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker4Range, newText: "'age'" },
+                },
+            ],
+        },
+        marker5: {
+            completions: [
+                {
+                    label: "'age'",
+                    kind: Consts.CompletionItemKind.Constant,
+                    textEdit: { range: marker5Range, newText: "'age'" },
+                },
+            ],
+        },
+    });
+
+    // @ts-ignore
+    await helper.verifyCompletion('excluded', 'markdown', {
+        marker3: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                },
+            ],
+        },
+        marker4: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                },
+            ],
+        },
+        marker5: {
+            completions: [
+                {
+                    label: "'name'",
+                    kind: Consts.CompletionItemKind.Constant,
+                },
+            ],
+        },
+    });
+}

package/src/tests/fourslash/completions.classVariable.fourslash.ts

@@ -0,0 +1,37 @@
+/// <reference path="fourslash.ts" />
+
+// @filename: test.py
+//// class B:
+////     var1 = 1
+////     var2: int
+////     var3: str = "hello"
+////     __var4 = 4
+////
+////     def __init__(self):
+////         self.var6 = 1
+////
+//// class T(B):
+////     var5: bool
+////     [|va/*marker*/|]
+
+// @ts-ignore
+await helper.verifyCompletion('included', 'markdown', {
+    marker: {
+        completions: [
+            {
+                label: 'var1',
+                kind: Consts.CompletionItemKind.Variable,
+            },
+            {
+                label: 'var2',
+                kind: Consts.CompletionItemKind.Variable,
+                textEdit: { range: helper.getPositionRange('marker'), newText: 'var2: int' },
+            },
+            {
+                label: 'var3',
+                kind: Consts.CompletionItemKind.Variable,
+                textEdit: { range: helper.getPositionRange('marker'), newText: 'var3: str' },
+            },
+        ],
+    },
+});

package/src/tests/fourslash/completions.comment.fourslash.ts

@@ -0,0 +1,28 @@
+/// <reference path="fourslash.ts" />
+
+// @filename: test.py
+//// msg = 'hello'
+//// [|/*marker1*/|]
+//// # msg.[|/*marker2*/|]
+//// [|/*marker3*/|]
+//// print('upper: ' + msg.up[|/*marker4*/|]per())
+//// print('#upper: ' + msg.up[|/*marker5*/|]per())
+////
+//// # msg.[|/*marker6*/|]
+//// [|/*marker7*/|]
+////
+
+// @ts-ignore
+await helper.verifyCompletion('exact', 'markdown', {
+    marker2: { completions: [] },
+    marker6: { completions: [] },
+});
+
+// @ts-ignore
+await helper.verifyCompletion('included', 'markdown', {
+    marker1: { completions: [{ label: 'msg', kind: Consts.CompletionItemKind.Variable }] },
+    marker3: { completions: [{ label: 'msg', kind: Consts.CompletionItemKind.Variable }] },
+    marker4: { completions: [{ label: 'upper', kind: Consts.CompletionItemKind.Method }] },
+    marker5: { completions: [{ label: 'upper', kind: Consts.CompletionItemKind.Method }] },
+    marker7: { completions: [{ label: 'msg', kind: Consts.CompletionItemKind.Variable }] },
+});

package/src/tests/fourslash/completions.declNames.class.fourslash.ts

@@ -0,0 +1,35 @@
+/// <reference path="fourslash.ts" />
+
+// @filename: test.py
+//// class [|/*marker1*/|]
+////
+
+// @filename: test1.py
+//// class c[|/*marker2*/|]
+////
+
+// @filename: test2.py
+//// class c1[|/*marker3*/|]():
+////     pass
+////
+
+// @filename: test3.py
+//// class c1([|/*marker4*/|]):
+////     pass
+////
+
+{
+    helper.openFiles(helper.getMarkers().map((m) => m.fileName));
+
+    // @ts-ignore
+    await helper.verifyCompletion('exact', 'markdown', {
+        marker1: { completions: [] },
+        marker2: { completions: [] },
+        marker3: { completions: [] },
+    });
+
+    // @ts-ignore
+    await helper.verifyCompletion('included', 'markdown', {
+        marker4: { completions: [{ label: 'Exception', kind: Consts.CompletionItemKind.Class }] },
+    });
+}

package/src/tests/fourslash/completions.declNames.exception.fourslash.ts

@@ -0,0 +1,18 @@
+/// <reference path="fourslash.ts" />
+
+// @filename: test.py
+//// try:
+////     pass
+//// except ZeroDivisionError as d[|/*marker1*/|]:
+////     pass
+////
+//// try:
+////     pass
+//// except ZeroDivisionError as [|/*marker2*/|]:
+////     pass
+
+// @ts-ignore
+await helper.verifyCompletion('exact', 'markdown', {
+    marker1: { completions: [] },
+    marker2: { completions: [] },
+});