@zzzen/pyright-internal 1.1.254 → 1.1.255

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (1265) hide show
  1. package/package.json +3 -2
  2. package/src/analyzer/aliasDeclarationUtils.ts +169 -0
  3. package/src/analyzer/analysis.ts +92 -0
  4. package/src/analyzer/analyzerFileInfo.ts +56 -0
  5. package/src/analyzer/analyzerNodeInfo.ts +198 -0
  6. package/src/analyzer/backgroundAnalysisProgram.ts +311 -0
  7. package/src/analyzer/binder.ts +4081 -0
  8. package/src/analyzer/checker.ts +5507 -0
  9. package/src/analyzer/circularDependency.ts +52 -0
  10. package/src/analyzer/codeFlowEngine.ts +1548 -0
  11. package/src/analyzer/codeFlowTypes.ts +266 -0
  12. package/src/analyzer/codeFlowUtils.ts +436 -0
  13. package/src/analyzer/commentUtils.ts +174 -0
  14. package/src/analyzer/constraintSolver.ts +926 -0
  15. package/src/analyzer/constructorTransform.ts +317 -0
  16. package/src/analyzer/dataClasses.ts +999 -0
  17. package/src/analyzer/declaration.ts +252 -0
  18. package/src/analyzer/declarationUtils.ts +238 -0
  19. package/src/analyzer/docStringConversion.ts +850 -0
  20. package/src/analyzer/docStringUtils.ts +112 -0
  21. package/src/analyzer/functionTransform.ts +137 -0
  22. package/src/analyzer/importResolver.ts +2308 -0
  23. package/src/analyzer/importResult.ts +98 -0
  24. package/src/analyzer/importStatementUtils.ts +834 -0
  25. package/src/analyzer/namedTuples.ts +422 -0
  26. package/src/analyzer/packageTypeReport.ts +108 -0
  27. package/src/analyzer/packageTypeVerifier.ts +1466 -0
  28. package/src/analyzer/parentDirectoryCache.ts +88 -0
  29. package/src/analyzer/parseTreeCleaner.ts +35 -0
  30. package/src/analyzer/parseTreeUtils.ts +2165 -0
  31. package/src/analyzer/parseTreeWalker.ts +663 -0
  32. package/src/analyzer/patternMatching.ts +1356 -0
  33. package/src/analyzer/program.ts +2778 -0
  34. package/src/analyzer/properties.ts +534 -0
  35. package/src/analyzer/protocols.ts +660 -0
  36. package/src/analyzer/pyTypedUtils.ts +52 -0
  37. package/src/analyzer/pythonPathUtils.ts +201 -0
  38. package/src/analyzer/scope.ts +202 -0
  39. package/src/analyzer/scopeUtils.ts +81 -0
  40. package/src/analyzer/service.ts +1679 -0
  41. package/src/analyzer/sourceFile.ts +1401 -0
  42. package/src/analyzer/sourceMapper.ts +717 -0
  43. package/src/analyzer/staticExpressions.ts +308 -0
  44. package/src/analyzer/symbol.ts +249 -0
  45. package/src/analyzer/symbolNameUtils.ts +50 -0
  46. package/src/analyzer/symbolUtils.ts +49 -0
  47. package/src/analyzer/testWalker.ts +101 -0
  48. package/src/analyzer/tracePrinter.ts +264 -0
  49. package/src/analyzer/typeCache.ts +181 -0
  50. package/src/analyzer/typeDocStringUtils.ts +371 -0
  51. package/src/analyzer/typeEvaluator.ts +23395 -0
  52. package/src/analyzer/typeEvaluatorTypes.ts +452 -0
  53. package/src/analyzer/typeEvaluatorWithTracker.ts +161 -0
  54. package/src/analyzer/typeGuards.ts +1828 -0
  55. package/src/analyzer/typePrinter.ts +807 -0
  56. package/src/analyzer/typeStubWriter.ts +727 -0
  57. package/src/analyzer/typeUtils.ts +3126 -0
  58. package/src/analyzer/typeVarContext.ts +330 -0
  59. package/src/analyzer/typedDicts.ts +950 -0
  60. package/src/analyzer/types.ts +2857 -0
  61. package/src/backgroundAnalysis.ts +49 -0
  62. package/src/backgroundAnalysisBase.ts +606 -0
  63. package/src/backgroundThreadBase.ts +143 -0
  64. package/src/commands/commandController.ts +62 -0
  65. package/src/commands/commandResult.ts +21 -0
  66. package/src/commands/commands.ts +16 -0
  67. package/src/commands/createTypeStub.ts +56 -0
  68. package/src/commands/quickActionCommand.ts +40 -0
  69. package/src/commands/restartServer.ts +20 -0
  70. package/src/common/cancellationUtils.ts +71 -0
  71. package/src/common/chokidarFileWatcherProvider.ts +74 -0
  72. package/src/common/collectionUtils.ts +379 -0
  73. package/src/common/commandLineOptions.ts +132 -0
  74. package/src/common/configOptions.ts +1291 -0
  75. package/src/common/console.ts +201 -0
  76. package/src/common/core.ts +152 -0
  77. package/src/common/crypto.ts +60 -0
  78. package/src/common/debug.ts +144 -0
  79. package/src/common/deferred.ts +78 -0
  80. package/src/common/diagnostic.ts +191 -0
  81. package/src/common/diagnosticRules.ts +80 -0
  82. package/src/common/diagnosticSink.ts +150 -0
  83. package/src/common/editAction.ts +46 -0
  84. package/src/common/extensibility.ts +35 -0
  85. package/src/common/extensions.ts +21 -0
  86. package/src/common/fileBasedCancellationUtils.ts +274 -0
  87. package/src/common/fileSystem.ts +152 -0
  88. package/src/common/fullAccessHost.ts +203 -0
  89. package/src/common/host.ts +49 -0
  90. package/src/common/logTracker.ts +157 -0
  91. package/src/common/lspUtils.ts +18 -0
  92. package/src/common/memUtils.ts +36 -0
  93. package/src/common/pathConsts.ts +16 -0
  94. package/src/common/pathUtils.ts +987 -0
  95. package/src/common/positionUtils.ts +78 -0
  96. package/src/common/progressReporter.ts +53 -0
  97. package/src/common/pythonVersion.ts +74 -0
  98. package/src/common/realFileSystem.ts +433 -0
  99. package/src/common/stringUtils.ts +161 -0
  100. package/src/common/textEditUtils.ts +31 -0
  101. package/src/common/textRange.ts +225 -0
  102. package/src/common/textRangeCollection.ts +122 -0
  103. package/src/common/timing.ts +105 -0
  104. package/src/common/uriParser.ts +26 -0
  105. package/src/common/workspaceEditUtils.ts +100 -0
  106. package/src/languageServerBase.ts +1556 -0
  107. package/src/languageService/analyzerServiceExecutor.ts +144 -0
  108. package/src/languageService/autoImporter.ts +926 -0
  109. package/src/languageService/callHierarchyProvider.ts +476 -0
  110. package/src/languageService/codeActionProvider.ts +85 -0
  111. package/src/languageService/completionProvider.ts +3103 -0
  112. package/src/languageService/definitionProvider.ts +235 -0
  113. package/src/languageService/documentHighlightProvider.ts +61 -0
  114. package/src/languageService/documentSymbolCollector.ts +466 -0
  115. package/src/languageService/documentSymbolProvider.ts +466 -0
  116. package/src/languageService/hoverProvider.ts +499 -0
  117. package/src/languageService/importAdder.ts +437 -0
  118. package/src/languageService/importSorter.ts +196 -0
  119. package/src/languageService/indentationUtils.ts +813 -0
  120. package/src/languageService/quickActions.ts +113 -0
  121. package/src/languageService/referencesProvider.ts +316 -0
  122. package/src/languageService/renameModuleProvider.ts +1400 -0
  123. package/src/languageService/signatureHelpProvider.ts +227 -0
  124. package/src/languageService/tooltipUtils.ts +147 -0
  125. package/src/localization/localize.ts +1272 -0
  126. package/src/localization/package.nls.de.json +3 -0
  127. package/src/localization/package.nls.en-us.json +636 -0
  128. package/src/localization/package.nls.es.json +3 -0
  129. package/src/localization/package.nls.fr.json +3 -0
  130. package/src/localization/package.nls.ja.json +3 -0
  131. package/src/localization/package.nls.ru.json +3 -0
  132. package/src/localization/package.nls.zh-cn.json +3 -0
  133. package/src/localization/package.nls.zh-tw.json +3 -0
  134. package/src/nodeMain.ts +21 -0
  135. package/src/nodeServer.ts +30 -0
  136. package/src/parser/characterStream.ts +145 -0
  137. package/src/parser/characters.ts +285 -0
  138. package/src/parser/parseNodes.ts +2338 -0
  139. package/src/parser/parser.ts +4859 -0
  140. package/src/parser/stringTokenUtils.ts +564 -0
  141. package/src/parser/tokenizer.ts +1379 -0
  142. package/src/parser/tokenizerTypes.ts +420 -0
  143. package/src/parser/unicode.ts +2917 -0
  144. package/src/pyright.ts +824 -0
  145. package/src/pyrightFileSystem.ts +237 -0
  146. package/src/readonlyAugmentedFileSystem.ts +204 -0
  147. package/src/server.ts +298 -0
  148. package/src/tests/chainedSourceFiles.test.ts +183 -0
  149. package/src/tests/checker.test.ts +445 -0
  150. package/src/tests/collectionUtils.test.ts +178 -0
  151. package/src/tests/common.test.ts +117 -0
  152. package/src/tests/config.test.ts +291 -0
  153. package/src/tests/debug.test.ts +120 -0
  154. package/src/tests/deferred.test.ts +54 -0
  155. package/src/tests/diagnosticOverrides.test.ts +104 -0
  156. package/src/tests/docStringConversion.test.ts +823 -0
  157. package/src/tests/docStringUtils.test.ts +57 -0
  158. package/src/tests/documentSymbolCollector.test.ts +735 -0
  159. package/src/tests/filesystem.test.ts +215 -0
  160. package/src/tests/fourSlashParser.test.ts +330 -0
  161. package/src/tests/fourSlashRunner.test.ts +36 -0
  162. package/src/tests/fourslash/completions.autoimport.Lib.Found.Type.fourslash.ts +33 -0
  163. package/src/tests/fourslash/completions.autoimport.Lib.Found.duplication.fourslash.ts +47 -0
  164. package/src/tests/fourslash/completions.autoimport.Lib.NotFound.fourslash.ts +14 -0
  165. package/src/tests/fourslash/completions.autoimport.disabled.fourslash.ts +25 -0
  166. package/src/tests/fourslash/completions.autoimport.duplicates.fourslash.ts +36 -0
  167. package/src/tests/fourslash/completions.autoimport.fourslash.ts +29 -0
  168. package/src/tests/fourslash/completions.autoimport.fromImport.fourslash.ts +130 -0
  169. package/src/tests/fourslash/completions.autoimport.plainText.fourslash.ts +29 -0
  170. package/src/tests/fourslash/completions.autoimport.shadow.fourslash.ts +48 -0
  171. package/src/tests/fourslash/completions.autoimport.submodule.fourslash.ts +31 -0
  172. package/src/tests/fourslash/completions.autoimport.topLevel.fourslash.ts +57 -0
  173. package/src/tests/fourslash/completions.builtinDocstrings.fourslash.ts +117 -0
  174. package/src/tests/fourslash/completions.builtinOverride.fourslash.ts +22 -0
  175. package/src/tests/fourslash/completions.call.stringLiteral.fourslash.ts +55 -0
  176. package/src/tests/fourslash/completions.call.typedDict.fourslash.ts +247 -0
  177. package/src/tests/fourslash/completions.call.typedDict.list.fourslash.ts +163 -0
  178. package/src/tests/fourslash/completions.call.typedDict.states.fourslash.ts +135 -0
  179. package/src/tests/fourslash/completions.classVariable.fourslash.ts +37 -0
  180. package/src/tests/fourslash/completions.comment.fourslash.ts +28 -0
  181. package/src/tests/fourslash/completions.declNames.class.fourslash.ts +35 -0
  182. package/src/tests/fourslash/completions.declNames.exception.fourslash.ts +18 -0
  183. package/src/tests/fourslash/completions.declNames.for.fourslash.ts +30 -0
  184. package/src/tests/fourslash/completions.declNames.importAlias.fourslash.ts +15 -0
  185. package/src/tests/fourslash/completions.declNames.lambda.fourslash.ts +45 -0
  186. package/src/tests/fourslash/completions.declNames.method.fourslash.ts +71 -0
  187. package/src/tests/fourslash/completions.declNames.overload.fourslash.ts +156 -0
  188. package/src/tests/fourslash/completions.declNames.topLevelOverload.fourslash.ts +141 -0
  189. package/src/tests/fourslash/completions.dictionary.keys.complex.fourslash.ts +76 -0
  190. package/src/tests/fourslash/completions.dictionary.keys.expression.fourslash.ts +63 -0
  191. package/src/tests/fourslash/completions.dictionary.keys.literalTypes.fourslash.ts +161 -0
  192. package/src/tests/fourslash/completions.dictionary.keys.simple.fourslash.ts +151 -0
  193. package/src/tests/fourslash/completions.dictionary.keys.stringLiterals.fourslash.ts +116 -0
  194. package/src/tests/fourslash/completions.dictionary.keys.symbols.fourslash.ts +53 -0
  195. package/src/tests/fourslash/completions.enums.members.fourslash.ts +44 -0
  196. package/src/tests/fourslash/completions.excluded.fourslash.ts +12 -0
  197. package/src/tests/fourslash/completions.fourslash.ts +51 -0
  198. package/src/tests/fourslash/completions.fstring.fourslash.ts +33 -0
  199. package/src/tests/fourslash/completions.fstring.stringLiteral.fourslash.ts +96 -0
  200. package/src/tests/fourslash/completions.fuzzyMatching.fourslash.ts +30 -0
  201. package/src/tests/fourslash/completions.importDunderNames.fourslash.ts +31 -0
  202. package/src/tests/fourslash/completions.importSubmodule.fourslash.ts +26 -0
  203. package/src/tests/fourslash/completions.importsDuplicates.fourslash.ts +23 -0
  204. package/src/tests/fourslash/completions.inList.fourslash.ts +33 -0
  205. package/src/tests/fourslash/completions.included.fourslash.ts +12 -0
  206. package/src/tests/fourslash/completions.indexer.keys.getitem.fourslash.ts +57 -0
  207. package/src/tests/fourslash/completions.inherited.function.docFromStub.fourslash.ts +28 -0
  208. package/src/tests/fourslash/completions.inherited.overload.docFromScrWithStub.fourslash.ts +52 -0
  209. package/src/tests/fourslash/completions.inherited.overload.docFromStub.fourslash.ts +53 -0
  210. package/src/tests/fourslash/completions.inherited.property.docFromSrc.fourslash.ts +104 -0
  211. package/src/tests/fourslash/completions.inherited.property.docFromStub.fourslash.ts +108 -0
  212. package/src/tests/fourslash/completions.keywords.pythonVersion.fourslash.ts +72 -0
  213. package/src/tests/fourslash/completions.libCodeAndStub.fourslash.ts +79 -0
  214. package/src/tests/fourslash/completions.libCodeNoStub.fourslash.ts +69 -0
  215. package/src/tests/fourslash/completions.libStub.fourslash.ts +69 -0
  216. package/src/tests/fourslash/completions.literals.fourslash.ts +33 -0
  217. package/src/tests/fourslash/completions.localCode.fourslash.ts +74 -0
  218. package/src/tests/fourslash/completions.moduleContext.UnknownMemberOnInstance.fourslash.ts +17 -0
  219. package/src/tests/fourslash/completions.moduleContext.UnknownStaticFunctionOnClass.fourslash.ts +57 -0
  220. package/src/tests/fourslash/completions.moduleContext.libCodeNoStub.fourslash.ts +30 -0
  221. package/src/tests/fourslash/completions.overloads.fourslash.ts +40 -0
  222. package/src/tests/fourslash/completions.override.default.fourslash.ts +47 -0
  223. package/src/tests/fourslash/completions.override.default.importStub.ts +50 -0
  224. package/src/tests/fourslash/completions.override.default.imported.fourslash.ts +61 -0
  225. package/src/tests/fourslash/completions.override.default.stub.fourslash.ts +47 -0
  226. package/src/tests/fourslash/completions.override.fourslash.ts +47 -0
  227. package/src/tests/fourslash/completions.override.property.fourslash.ts +31 -0
  228. package/src/tests/fourslash/completions.override.property.stub.fourslash.ts +31 -0
  229. package/src/tests/fourslash/completions.override.staticAndClassmethod.fourslash.ts +64 -0
  230. package/src/tests/fourslash/completions.override.stub.fourslash.ts +47 -0
  231. package/src/tests/fourslash/completions.override2.fourslash.ts +71 -0
  232. package/src/tests/fourslash/completions.parameters.fourslash.ts +23 -0
  233. package/src/tests/fourslash/completions.params.fourslash.ts +24 -0
  234. package/src/tests/fourslash/completions.parentFolder.fourslash.ts +17 -0
  235. package/src/tests/fourslash/completions.parentFolders.fourslash.ts +77 -0
  236. package/src/tests/fourslash/completions.plainText.fourslash.ts +34 -0
  237. package/src/tests/fourslash/completions.private.fourslash.ts +64 -0
  238. package/src/tests/fourslash/completions.property.fourslash.ts +28 -0
  239. package/src/tests/fourslash/completions.propertyDocStrings.fourslash.ts +64 -0
  240. package/src/tests/fourslash/completions.self.fourslash.ts +43 -0
  241. package/src/tests/fourslash/completions.stringLiteral.escape.fourslash.ts +71 -0
  242. package/src/tests/fourslash/completions.stringLiteral.fourslash.ts +79 -0
  243. package/src/tests/fourslash/completions.typeAlias.fourslash.ts +29 -0
  244. package/src/tests/fourslash/completions.typeshed.fourslash.ts +9 -0
  245. package/src/tests/fourslash/completions.vardecls.fourslash.ts +40 -0
  246. package/src/tests/fourslash/completions.variableDocStrings.fourslash.ts +69 -0
  247. package/src/tests/fourslash/completions.wildcardimports.fourslash.ts +116 -0
  248. package/src/tests/fourslash/completions.with.fourslash.ts +26 -0
  249. package/src/tests/fourslash/diagnostics.missingModuleSource.fourslash.ts +69 -0
  250. package/src/tests/fourslash/findDefinitions.builtinClass.fourslash.ts +29 -0
  251. package/src/tests/fourslash/findDefinitions.classes.fourslash.ts +101 -0
  252. package/src/tests/fourslash/findDefinitions.definitionFilter.preferSource.fourslash.ts +29 -0
  253. package/src/tests/fourslash/findDefinitions.definitionFilter.preferSource.onlyStubs.ts +24 -0
  254. package/src/tests/fourslash/findDefinitions.definitionFilter.preferStub.fourslash.ts +29 -0
  255. package/src/tests/fourslash/findDefinitions.definitionFilter.preferStub.onlySource.fourslash.ts +25 -0
  256. package/src/tests/fourslash/findDefinitions.fields.fourslash.ts +131 -0
  257. package/src/tests/fourslash/findDefinitions.functions.fourslash.ts +87 -0
  258. package/src/tests/fourslash/findDefinitions.methods.fourslash.ts +162 -0
  259. package/src/tests/fourslash/findDefinitions.modules.fourslash.ts +50 -0
  260. package/src/tests/fourslash/findDefinitions.namespaceImport.fourslash.ts +40 -0
  261. package/src/tests/fourslash/findDefinitions.overloads.fourslash.ts +223 -0
  262. package/src/tests/fourslash/findDefinitions.parameters.fourslash.ts +162 -0
  263. package/src/tests/fourslash/findDefinitions.sourceAndStub.function.fourslash.ts +26 -0
  264. package/src/tests/fourslash/findDefinitions.sourceAndStub.innerClass.fourslash.ts +32 -0
  265. package/src/tests/fourslash/findDefinitions.sourceAndStub.innerClassMethod.fourslash.ts +33 -0
  266. package/src/tests/fourslash/findDefinitions.sourceAndStub.outerClass.fourslash.ts +28 -0
  267. package/src/tests/fourslash/findDefinitions.sourceAndStub.outerClassMethod.fourslash.ts +29 -0
  268. package/src/tests/fourslash/findDefinitions.sourceAndStub.outerClassPropertyReadOnly.fourslash.ts +31 -0
  269. package/src/tests/fourslash/findDefinitions.sourceAndStub.outerClassPropertyReadWrite.fourslash.ts +36 -0
  270. package/src/tests/fourslash/findDefinitions.sourceOnly.class.fourslash.ts +29 -0
  271. package/src/tests/fourslash/findDefinitions.sourceOnly.function1.fourslash.ts +28 -0
  272. package/src/tests/fourslash/findDefinitions.sourceOnly.function2.fourslash.ts +28 -0
  273. package/src/tests/fourslash/findDefinitions.sourceOnly.relativeImport1.fourslash.ts +22 -0
  274. package/src/tests/fourslash/findDefinitions.sourceOnly.relativeImport2.fourslash.ts +22 -0
  275. package/src/tests/fourslash/findDefinitions.stubOnly.fourslash.ts +61 -0
  276. package/src/tests/fourslash/findDefinitions.stubPackages.fourslash.ts +71 -0
  277. package/src/tests/fourslash/findDefinitions.typedDict.keys.fourslash.ts +99 -0
  278. package/src/tests/fourslash/findDefinitions.variables.fourslash.ts +88 -0
  279. package/src/tests/fourslash/findDefinitions.wildcardimports.fourslash.ts +117 -0
  280. package/src/tests/fourslash/findTypeDefinitions.builtinClass.fourslash.ts +26 -0
  281. package/src/tests/fourslash/findTypeDefinitions.classes.fourslash.ts +72 -0
  282. package/src/tests/fourslash/findTypeDefinitions.unions.fourslash.ts +26 -0
  283. package/src/tests/fourslash/findallreferences.classPropertyReadWrite.ts +39 -0
  284. package/src/tests/fourslash/findallreferences.fourslash.ts +34 -0
  285. package/src/tests/fourslash/findallreferences.importalias.fourslash.ts +34 -0
  286. package/src/tests/fourslash/findallreferences.invokedFromLibrary.fourslash.ts +52 -0
  287. package/src/tests/fourslash/findallreferences.module.nested.fourslash.ts +67 -0
  288. package/src/tests/fourslash/findallreferences.modules.duplicated.fourslash.ts +59 -0
  289. package/src/tests/fourslash/findallreferences.modules.fourslash.ts +46 -0
  290. package/src/tests/fourslash/findallreferences.modules.shadow.fourslash.ts +95 -0
  291. package/src/tests/fourslash/findallreferences.openFiles.fourslash.ts +37 -0
  292. package/src/tests/fourslash/findallreferences.parameter.fourslash.ts +24 -0
  293. package/src/tests/fourslash/findallreferences.sourceAndStub.class.fourslash.ts +33 -0
  294. package/src/tests/fourslash/findallreferences.sourceAndStub.classMethod.fourslash.ts +37 -0
  295. package/src/tests/fourslash/findallreferences.sourceAndStub.classPropertyReadOnly.fourslash.ts +40 -0
  296. package/src/tests/fourslash/findallreferences.sourceAndStub.classPropertyReadWrite.fourslash.skip.ts +48 -0
  297. package/src/tests/fourslash/findallreferences.sourceAndStub.function.fourslash.ts +35 -0
  298. package/src/tests/fourslash/findallreferences.variable.fourslash.ts +24 -0
  299. package/src/tests/fourslash/fourslash.ts +411 -0
  300. package/src/tests/fourslash/highlightreferences.attributes.fourslash.ts +31 -0
  301. package/src/tests/fourslash/hover.builtinDocstrings.fourslash.ts +64 -0
  302. package/src/tests/fourslash/hover.classNoInit.fourslash.ts +14 -0
  303. package/src/tests/fourslash/hover.docFromScr.stringFormat.fourslash.ts +52 -0
  304. package/src/tests/fourslash/hover.docFromSrc.fourslash.ts +83 -0
  305. package/src/tests/fourslash/hover.docFromSrc.pkg-vs-module1.fourslash.ts +26 -0
  306. package/src/tests/fourslash/hover.docFromSrc.pkg-vs-module2.fourslash.ts +26 -0
  307. package/src/tests/fourslash/hover.docFromSrc.relativeImport1.fourslash.ts +23 -0
  308. package/src/tests/fourslash/hover.docFromSrc.relativeImport2.fourslash.ts +22 -0
  309. package/src/tests/fourslash/hover.docFromSrc.relativeImport3.fourslash.ts +46 -0
  310. package/src/tests/fourslash/hover.docFromSrc.stubs-package.fourslash.ts +29 -0
  311. package/src/tests/fourslash/hover.docFromSrc.typeshed.fourslash.ts +20 -0
  312. package/src/tests/fourslash/hover.docstring.links.fourslash.ts +12 -0
  313. package/src/tests/fourslash/hover.docstring.overloads.fourslash.ts +41 -0
  314. package/src/tests/fourslash/hover.docstring.split.fourslash.ts +24 -0
  315. package/src/tests/fourslash/hover.fourslash.ts +20 -0
  316. package/src/tests/fourslash/hover.import.django.view.fourslash.ts +27 -0
  317. package/src/tests/fourslash/hover.import.fourslash.ts +11 -0
  318. package/src/tests/fourslash/hover.inherited.docFromSrc.fourslash.ts +63 -0
  319. package/src/tests/fourslash/hover.inherited.docFromSrcWithStub.fourslash.ts +54 -0
  320. package/src/tests/fourslash/hover.inherited.docFromStub.fourslash.ts +42 -0
  321. package/src/tests/fourslash/hover.inherited.overload.docFromSrcWithStub.fourslash.ts +46 -0
  322. package/src/tests/fourslash/hover.inherited.overload.docFromStub.fourslash.ts +47 -0
  323. package/src/tests/fourslash/hover.inherited.property.docFromSrcWithStub.fourslash.ts +87 -0
  324. package/src/tests/fourslash/hover.inherited.property.docFromStub.fourslash.ts +91 -0
  325. package/src/tests/fourslash/hover.init.fourslash.ts +39 -0
  326. package/src/tests/fourslash/hover.libCodeAndStub.fourslash.ts +52 -0
  327. package/src/tests/fourslash/hover.libCodeNoStub.fourslash.ts +41 -0
  328. package/src/tests/fourslash/hover.libStub.fourslash.ts +41 -0
  329. package/src/tests/fourslash/hover.optionalAliasParameter.fourslash.ts +14 -0
  330. package/src/tests/fourslash/hover.plainText.fourslash.ts +21 -0
  331. package/src/tests/fourslash/hover.typedDict.key.fourslash.ts +51 -0
  332. package/src/tests/fourslash/hover.variable.docString.fourslash.ts +41 -0
  333. package/src/tests/fourslash/hover.wildcardimports.fourslash.ts +80 -0
  334. package/src/tests/fourslash/import.publicSymbols.fourslash.ts +65 -0
  335. package/src/tests/fourslash/import.pytyped.dunderAll.fourslash.ts +77 -0
  336. package/src/tests/fourslash/import.pytyped.privateSymbols.fourslash.ts +65 -0
  337. package/src/tests/fourslash/import.pytyped.typeCheckingBasic.fourslash.ts +38 -0
  338. package/src/tests/fourslash/import.pytyped.typeCheckingOff.fourslash.ts +38 -0
  339. package/src/tests/fourslash/import.wildcard.fourslash.ts +27 -0
  340. package/src/tests/fourslash/importnotresolved.fourslash.ts +13 -0
  341. package/src/tests/fourslash/missingModuleSource.disablingInStrictMode.fourslash.ts +20 -0
  342. package/src/tests/fourslash/missingModuleSource.fourslash.ts +14 -0
  343. package/src/tests/fourslash/missingTypeStub.codeAction.fourslash.ts +33 -0
  344. package/src/tests/fourslash/missingTypeStub.command.multipart.fourslash.ts +48 -0
  345. package/src/tests/fourslash/missingTypeStub.command.singlefile.fourslash.ts +38 -0
  346. package/src/tests/fourslash/missingTypeStub.command.singlepart.fourslash.ts +38 -0
  347. package/src/tests/fourslash/missingTypeStub.fourslash.ts +20 -0
  348. package/src/tests/fourslash/missingTypeStub.invokeCodeAction.fourslash.ts +36 -0
  349. package/src/tests/fourslash/noerrors.fourslash.ts +9 -0
  350. package/src/tests/fourslash/orderImports1.command.fourslash.ts +20 -0
  351. package/src/tests/fourslash/orderImports2.command.fourslash.ts +24 -0
  352. package/src/tests/fourslash/rename.externallyHidden.fourslash.ts +24 -0
  353. package/src/tests/fourslash/rename.externallyHidden.params.fourslash.ts +24 -0
  354. package/src/tests/fourslash/rename.fourslash.ts +29 -0
  355. package/src/tests/fourslash/rename.library.fourslash.ts +29 -0
  356. package/src/tests/fourslash/rename.library.sourceAndStub.fourslash.ts +33 -0
  357. package/src/tests/fourslash/rename.multipleDecl.fourslash.ts +21 -0
  358. package/src/tests/fourslash/rename.sourceAndStub.fourslash.ts +37 -0
  359. package/src/tests/fourslash/rename.string.excluded.fourslash.ts +43 -0
  360. package/src/tests/fourslash/rename.string.fourslash.ts +26 -0
  361. package/src/tests/fourslash/signature.builtinDocstrings.fourslash.ts +63 -0
  362. package/src/tests/fourslash/signature.complicated.fourslash.ts +100 -0
  363. package/src/tests/fourslash/signature.cornercases.fourslash.ts +23 -0
  364. package/src/tests/fourslash/signature.docstrings.fourslash.ts +43 -0
  365. package/src/tests/fourslash/signature.docstrings.overloaded.fourslash.ts +50 -0
  366. package/src/tests/fourslash/signature.docstrings.wildcardimports.fourslash.ts +122 -0
  367. package/src/tests/fourslash/signature.overload.fourslash.ts +64 -0
  368. package/src/tests/fourslash/signature.simple.fourslash.ts +104 -0
  369. package/src/tests/harness/fourslash/fourSlashParser.ts +441 -0
  370. package/src/tests/harness/fourslash/fourSlashTypes.ts +134 -0
  371. package/src/tests/harness/fourslash/runner.ts +99 -0
  372. package/src/tests/harness/fourslash/testLanguageService.ts +142 -0
  373. package/src/tests/harness/fourslash/testState.Consts.ts +29 -0
  374. package/src/tests/harness/fourslash/testState.ts +1993 -0
  375. package/src/tests/harness/testAccessHost.ts +23 -0
  376. package/src/tests/harness/testHost.ts +177 -0
  377. package/src/tests/harness/utils.ts +355 -0
  378. package/src/tests/harness/vfs/factory.ts +199 -0
  379. package/src/tests/harness/vfs/filesystem.ts +1893 -0
  380. package/src/tests/harness/vfs/pathValidation.ts +154 -0
  381. package/src/tests/importAdder.test.ts +1361 -0
  382. package/src/tests/importResolver.test.ts +534 -0
  383. package/src/tests/importStatementUtils.test.ts +526 -0
  384. package/src/tests/indentationUtils.ptvs.test.ts +388 -0
  385. package/src/tests/indentationUtils.reindent.test.ts +403 -0
  386. package/src/tests/indentationUtils.test.ts +455 -0
  387. package/src/tests/ipythonMode.test.ts +365 -0
  388. package/src/tests/localizer.test.ts +47 -0
  389. package/src/tests/parseTreeUtils.test.ts +281 -0
  390. package/src/tests/parser.test.ts +99 -0
  391. package/src/tests/pathUtils.test.ts +321 -0
  392. package/src/tests/pyrightFileSystem.test.ts +183 -0
  393. package/src/tests/renameModule.folder.test.ts +261 -0
  394. package/src/tests/renameModule.fromImports.test.ts +952 -0
  395. package/src/tests/renameModule.imports.test.ts +450 -0
  396. package/src/tests/renameModule.misc.test.ts +765 -0
  397. package/src/tests/renameModule.relativePath.test.ts +277 -0
  398. package/src/tests/renameModuleTestUtils.ts +210 -0
  399. package/src/tests/samples/abstractClass1.py +51 -0
  400. package/src/tests/samples/abstractClass2.py +25 -0
  401. package/src/tests/samples/abstractClass3.py +30 -0
  402. package/src/tests/samples/abstractClass4.py +30 -0
  403. package/src/tests/samples/abstractClass5.py +42 -0
  404. package/src/tests/samples/abstractClass6.py +25 -0
  405. package/src/tests/samples/abstractClass7.py +22 -0
  406. package/src/tests/samples/abstractClass8.py +27 -0
  407. package/src/tests/samples/annotated1.py +72 -0
  408. package/src/tests/samples/annotatedVar1.py +22 -0
  409. package/src/tests/samples/annotatedVar2.py +28 -0
  410. package/src/tests/samples/annotatedVar3.py +54 -0
  411. package/src/tests/samples/annotatedVar4.py +24 -0
  412. package/src/tests/samples/annotatedVar5.py +37 -0
  413. package/src/tests/samples/annotatedVar6.py +22 -0
  414. package/src/tests/samples/annotatedVar7.py +13 -0
  415. package/src/tests/samples/annotatedVar8.py +29 -0
  416. package/src/tests/samples/annotations1.py +98 -0
  417. package/src/tests/samples/annotations2.py +33 -0
  418. package/src/tests/samples/annotations3.py +43 -0
  419. package/src/tests/samples/annotations4.py +54 -0
  420. package/src/tests/samples/annotations5.py +9 -0
  421. package/src/tests/samples/annotations6.py +29 -0
  422. package/src/tests/samples/assert1.py +22 -0
  423. package/src/tests/samples/assertType1.py +46 -0
  424. package/src/tests/samples/assignment1.py +58 -0
  425. package/src/tests/samples/assignment10.py +14 -0
  426. package/src/tests/samples/assignment2.py +40 -0
  427. package/src/tests/samples/assignment3.py +46 -0
  428. package/src/tests/samples/assignment4.py +20 -0
  429. package/src/tests/samples/assignment5.py +16 -0
  430. package/src/tests/samples/assignment6.py +25 -0
  431. package/src/tests/samples/assignment7.py +13 -0
  432. package/src/tests/samples/assignment8.py +40 -0
  433. package/src/tests/samples/assignment9.py +25 -0
  434. package/src/tests/samples/assignmentExpr1.py +23 -0
  435. package/src/tests/samples/assignmentExpr2.py +49 -0
  436. package/src/tests/samples/assignmentExpr3.py +27 -0
  437. package/src/tests/samples/assignmentExpr4.py +49 -0
  438. package/src/tests/samples/assignmentExpr5.py +15 -0
  439. package/src/tests/samples/assignmentExpr6.py +10 -0
  440. package/src/tests/samples/assignmentExpr7.py +13 -0
  441. package/src/tests/samples/assignmentExpr8.py +11 -0
  442. package/src/tests/samples/assignmentExpr9.py +28 -0
  443. package/src/tests/samples/async1.py +49 -0
  444. package/src/tests/samples/augmentedAssignment1.py +62 -0
  445. package/src/tests/samples/augmentedAssignment2.py +23 -0
  446. package/src/tests/samples/augmentedAssignment3.py +8 -0
  447. package/src/tests/samples/await1.py +23 -0
  448. package/src/tests/samples/await2.py +25 -0
  449. package/src/tests/samples/badToken1.py +6 -0
  450. package/src/tests/samples/builtins1.py +2 -0
  451. package/src/tests/samples/call1.py +93 -0
  452. package/src/tests/samples/call2.py +118 -0
  453. package/src/tests/samples/call3.py +168 -0
  454. package/src/tests/samples/call4.py +13 -0
  455. package/src/tests/samples/call5.py +95 -0
  456. package/src/tests/samples/call6.py +35 -0
  457. package/src/tests/samples/call7.py +57 -0
  458. package/src/tests/samples/callSite1.py +6 -0
  459. package/src/tests/samples/callSite2.py +12 -0
  460. package/src/tests/samples/callable1.py +53 -0
  461. package/src/tests/samples/callable2.py +58 -0
  462. package/src/tests/samples/callable3.py +28 -0
  463. package/src/tests/samples/callable4.py +32 -0
  464. package/src/tests/samples/callable5.py +40 -0
  465. package/src/tests/samples/callable6.py +83 -0
  466. package/src/tests/samples/callbackProtocol1.py +119 -0
  467. package/src/tests/samples/callbackProtocol2.py +29 -0
  468. package/src/tests/samples/callbackProtocol3.py +23 -0
  469. package/src/tests/samples/callbackProtocol4.py +73 -0
  470. package/src/tests/samples/callbackProtocol5.py +66 -0
  471. package/src/tests/samples/callbackProtocol6.py +66 -0
  472. package/src/tests/samples/callbackProtocol7.py +16 -0
  473. package/src/tests/samples/callbackProtocol8.py +16 -0
  474. package/src/tests/samples/capturedVariable1.py +106 -0
  475. package/src/tests/samples/circular1.py +16 -0
  476. package/src/tests/samples/circular2.py +38 -0
  477. package/src/tests/samples/circularBaseClass.py +17 -0
  478. package/src/tests/samples/classGetItem1.py +30 -0
  479. package/src/tests/samples/classVar1.py +41 -0
  480. package/src/tests/samples/classVar2.py +29 -0
  481. package/src/tests/samples/classVar3.py +46 -0
  482. package/src/tests/samples/classVar4.py +36 -0
  483. package/src/tests/samples/classes1.py +50 -0
  484. package/src/tests/samples/classes3.py +48 -0
  485. package/src/tests/samples/classes4.py +15 -0
  486. package/src/tests/samples/classes5.py +260 -0
  487. package/src/tests/samples/classes6.py +16 -0
  488. package/src/tests/samples/classes7.py +18 -0
  489. package/src/tests/samples/classes8.py +47 -0
  490. package/src/tests/samples/codeFlow1.py +65 -0
  491. package/src/tests/samples/codeFlow2.py +26 -0
  492. package/src/tests/samples/codeFlow3.py +8 -0
  493. package/src/tests/samples/codeFlow4.py +133 -0
  494. package/src/tests/samples/codeFlow5.py +10 -0
  495. package/src/tests/samples/codeFlow6.py +35 -0
  496. package/src/tests/samples/codeFlow7.py +35 -0
  497. package/src/tests/samples/comparison1.py +77 -0
  498. package/src/tests/samples/comparison2.py +37 -0
  499. package/src/tests/samples/complex1.py +17 -0
  500. package/src/tests/samples/constant1.py +40 -0
  501. package/src/tests/samples/constants1.py +64 -0
  502. package/src/tests/samples/constructor1.py +30 -0
  503. package/src/tests/samples/constructor10.py +19 -0
  504. package/src/tests/samples/constructor11.py +32 -0
  505. package/src/tests/samples/constructor12.py +20 -0
  506. package/src/tests/samples/constructor13.py +15 -0
  507. package/src/tests/samples/constructor2.py +180 -0
  508. package/src/tests/samples/constructor3.py +18 -0
  509. package/src/tests/samples/constructor4.py +25 -0
  510. package/src/tests/samples/constructor5.py +25 -0
  511. package/src/tests/samples/constructor6.py +70 -0
  512. package/src/tests/samples/constructor7.py +11 -0
  513. package/src/tests/samples/constructor8.py +99 -0
  514. package/src/tests/samples/constructor9.py +19 -0
  515. package/src/tests/samples/coroutines1.py +60 -0
  516. package/src/tests/samples/coroutines2.py +20 -0
  517. package/src/tests/samples/coroutines3.py +37 -0
  518. package/src/tests/samples/dataclass1.py +44 -0
  519. package/src/tests/samples/dataclass10.py +45 -0
  520. package/src/tests/samples/dataclass11.py +53 -0
  521. package/src/tests/samples/dataclass12.py +32 -0
  522. package/src/tests/samples/dataclass13.py +42 -0
  523. package/src/tests/samples/dataclass14.py +38 -0
  524. package/src/tests/samples/dataclass15.py +58 -0
  525. package/src/tests/samples/dataclass16.py +21 -0
  526. package/src/tests/samples/dataclass17.py +46 -0
  527. package/src/tests/samples/dataclass18.py +57 -0
  528. package/src/tests/samples/dataclass19.py +44 -0
  529. package/src/tests/samples/dataclass2.py +30 -0
  530. package/src/tests/samples/dataclass20.py +71 -0
  531. package/src/tests/samples/dataclass21.py +13 -0
  532. package/src/tests/samples/dataclass3.py +18 -0
  533. package/src/tests/samples/dataclass4.py +64 -0
  534. package/src/tests/samples/dataclass5.py +81 -0
  535. package/src/tests/samples/dataclass6.py +29 -0
  536. package/src/tests/samples/dataclass7.py +93 -0
  537. package/src/tests/samples/dataclass8.py +22 -0
  538. package/src/tests/samples/dataclass9.py +24 -0
  539. package/src/tests/samples/dataclassPostInit1.py +61 -0
  540. package/src/tests/samples/dataclassTransform1.py +71 -0
  541. package/src/tests/samples/dataclassTransform2.py +78 -0
  542. package/src/tests/samples/dataclassTransform3.py +83 -0
  543. package/src/tests/samples/dataclassTransform4.py +87 -0
  544. package/src/tests/samples/decorator1.py +20 -0
  545. package/src/tests/samples/decorator2.py +31 -0
  546. package/src/tests/samples/decorator3.py +27 -0
  547. package/src/tests/samples/decorator4.py +37 -0
  548. package/src/tests/samples/decorator5.py +29 -0
  549. package/src/tests/samples/decorator6.py +34 -0
  550. package/src/tests/samples/defaultInitializer1.py +26 -0
  551. package/src/tests/samples/deprecated1.py +37 -0
  552. package/src/tests/samples/descriptor1.py +134 -0
  553. package/src/tests/samples/descriptor2.py +29 -0
  554. package/src/tests/samples/dictionary1.py +42 -0
  555. package/src/tests/samples/dictionary2.py +14 -0
  556. package/src/tests/samples/dictionary3.py +30 -0
  557. package/src/tests/samples/dictionary4.py +106 -0
  558. package/src/tests/samples/dunderAll1.py +32 -0
  559. package/src/tests/samples/dunderAll2.py +16 -0
  560. package/src/tests/samples/dunderAll3.pyi +16 -0
  561. package/src/tests/samples/duplicateDeclaration1.py +115 -0
  562. package/src/tests/samples/duplicateDeclaration2.py +45 -0
  563. package/src/tests/samples/duplicateImports1.py +13 -0
  564. package/src/tests/samples/ellipsis1.pyi +48 -0
  565. package/src/tests/samples/emptyContainers1.py +89 -0
  566. package/src/tests/samples/enums1.py +52 -0
  567. package/src/tests/samples/enums2.py +15 -0
  568. package/src/tests/samples/enums3.py +24 -0
  569. package/src/tests/samples/enums4.py +23 -0
  570. package/src/tests/samples/enums5.py +16 -0
  571. package/src/tests/samples/enums6.py +33 -0
  572. package/src/tests/samples/enums7.py +54 -0
  573. package/src/tests/samples/enums8.py +14 -0
  574. package/src/tests/samples/expressions1.py +54 -0
  575. package/src/tests/samples/expressions2.py +29 -0
  576. package/src/tests/samples/expressions3.py +26 -0
  577. package/src/tests/samples/expressions4.py +23 -0
  578. package/src/tests/samples/expressions5.py +70 -0
  579. package/src/tests/samples/expressions6.py +14 -0
  580. package/src/tests/samples/expressions7.py +40 -0
  581. package/src/tests/samples/expressions8.py +26 -0
  582. package/src/tests/samples/expressions9.py +14 -0
  583. package/src/tests/samples/final1.py +14 -0
  584. package/src/tests/samples/final2.py +86 -0
  585. package/src/tests/samples/final3.py +179 -0
  586. package/src/tests/samples/final4.pyi +21 -0
  587. package/src/tests/samples/final5.py +18 -0
  588. package/src/tests/samples/forLoop1.py +88 -0
  589. package/src/tests/samples/forLoop2.py +48 -0
  590. package/src/tests/samples/fstring1.py +66 -0
  591. package/src/tests/samples/fstring2.py +12 -0
  592. package/src/tests/samples/fstring3.py +35 -0
  593. package/src/tests/samples/fstring4.py +17 -0
  594. package/src/tests/samples/fstring5.py +17 -0
  595. package/src/tests/samples/fstring6.py +14 -0
  596. package/src/tests/samples/function1.py +12 -0
  597. package/src/tests/samples/function10.py +20 -0
  598. package/src/tests/samples/function11.py +36 -0
  599. package/src/tests/samples/function2.py +7 -0
  600. package/src/tests/samples/function3.py +82 -0
  601. package/src/tests/samples/function4.py +13 -0
  602. package/src/tests/samples/function6.py +24 -0
  603. package/src/tests/samples/function7.py +13 -0
  604. package/src/tests/samples/function8.py +38 -0
  605. package/src/tests/samples/function9.py +75 -0
  606. package/src/tests/samples/functionAnnotation1.py +61 -0
  607. package/src/tests/samples/functionAnnotation2.py +26 -0
  608. package/src/tests/samples/functionAnnotation3.py +37 -0
  609. package/src/tests/samples/functionAnnotation4.py +22 -0
  610. package/src/tests/samples/functionMember1.py +18 -0
  611. package/src/tests/samples/functionMember2.py +45 -0
  612. package/src/tests/samples/generators1.py +122 -0
  613. package/src/tests/samples/generators10.py +18 -0
  614. package/src/tests/samples/generators11.py +27 -0
  615. package/src/tests/samples/generators12.py +25 -0
  616. package/src/tests/samples/generators13.py +65 -0
  617. package/src/tests/samples/generators14.py +19 -0
  618. package/src/tests/samples/generators15.py +26 -0
  619. package/src/tests/samples/generators2.py +31 -0
  620. package/src/tests/samples/generators3.py +42 -0
  621. package/src/tests/samples/generators4.py +33 -0
  622. package/src/tests/samples/generators5.py +24 -0
  623. package/src/tests/samples/generators6.py +14 -0
  624. package/src/tests/samples/generators7.py +18 -0
  625. package/src/tests/samples/generators8.py +12 -0
  626. package/src/tests/samples/generators9.py +19 -0
  627. package/src/tests/samples/generic1.py +41 -0
  628. package/src/tests/samples/genericTypes1.py +12 -0
  629. package/src/tests/samples/genericTypes10.py +38 -0
  630. package/src/tests/samples/genericTypes11.py +26 -0
  631. package/src/tests/samples/genericTypes12.py +14 -0
  632. package/src/tests/samples/genericTypes13.py +27 -0
  633. package/src/tests/samples/genericTypes14.py +22 -0
  634. package/src/tests/samples/genericTypes15.py +13 -0
  635. package/src/tests/samples/genericTypes16.py +15 -0
  636. package/src/tests/samples/genericTypes17.py +24 -0
  637. package/src/tests/samples/genericTypes18.py +135 -0
  638. package/src/tests/samples/genericTypes19.py +32 -0
  639. package/src/tests/samples/genericTypes2.py +40 -0
  640. package/src/tests/samples/genericTypes20.py +44 -0
  641. package/src/tests/samples/genericTypes21.py +29 -0
  642. package/src/tests/samples/genericTypes22.py +23 -0
  643. package/src/tests/samples/genericTypes23.py +27 -0
  644. package/src/tests/samples/genericTypes24.py +32 -0
  645. package/src/tests/samples/genericTypes25.py +37 -0
  646. package/src/tests/samples/genericTypes26.py +31 -0
  647. package/src/tests/samples/genericTypes27.py +14 -0
  648. package/src/tests/samples/genericTypes28.py +51 -0
  649. package/src/tests/samples/genericTypes29.py +24 -0
  650. package/src/tests/samples/genericTypes3.py +13 -0
  651. package/src/tests/samples/genericTypes30.py +32 -0
  652. package/src/tests/samples/genericTypes31.py +30 -0
  653. package/src/tests/samples/genericTypes32.py +25 -0
  654. package/src/tests/samples/genericTypes33.py +46 -0
  655. package/src/tests/samples/genericTypes34.py +54 -0
  656. package/src/tests/samples/genericTypes35.py +45 -0
  657. package/src/tests/samples/genericTypes36.py +35 -0
  658. package/src/tests/samples/genericTypes37.py +26 -0
  659. package/src/tests/samples/genericTypes38.py +15 -0
  660. package/src/tests/samples/genericTypes39.py +82 -0
  661. package/src/tests/samples/genericTypes4.py +37 -0
  662. package/src/tests/samples/genericTypes40.py +38 -0
  663. package/src/tests/samples/genericTypes41.py +16 -0
  664. package/src/tests/samples/genericTypes42.py +32 -0
  665. package/src/tests/samples/genericTypes43.py +24 -0
  666. package/src/tests/samples/genericTypes44.py +33 -0
  667. package/src/tests/samples/genericTypes45.py +43 -0
  668. package/src/tests/samples/genericTypes46.py +66 -0
  669. package/src/tests/samples/genericTypes47.py +39 -0
  670. package/src/tests/samples/genericTypes48.py +18 -0
  671. package/src/tests/samples/genericTypes49.py +41 -0
  672. package/src/tests/samples/genericTypes5.py +42 -0
  673. package/src/tests/samples/genericTypes50.py +62 -0
  674. package/src/tests/samples/genericTypes51.py +55 -0
  675. package/src/tests/samples/genericTypes52.py +34 -0
  676. package/src/tests/samples/genericTypes53.py +42 -0
  677. package/src/tests/samples/genericTypes54.py +31 -0
  678. package/src/tests/samples/genericTypes55.py +61 -0
  679. package/src/tests/samples/genericTypes56.py +19 -0
  680. package/src/tests/samples/genericTypes57.py +30 -0
  681. package/src/tests/samples/genericTypes58.py +52 -0
  682. package/src/tests/samples/genericTypes59.py +71 -0
  683. package/src/tests/samples/genericTypes6.py +49 -0
  684. package/src/tests/samples/genericTypes60.py +27 -0
  685. package/src/tests/samples/genericTypes61.py +34 -0
  686. package/src/tests/samples/genericTypes62.py +26 -0
  687. package/src/tests/samples/genericTypes63.py +32 -0
  688. package/src/tests/samples/genericTypes64.py +16 -0
  689. package/src/tests/samples/genericTypes65.py +15 -0
  690. package/src/tests/samples/genericTypes66.py +30 -0
  691. package/src/tests/samples/genericTypes67.py +57 -0
  692. package/src/tests/samples/genericTypes68.py +14 -0
  693. package/src/tests/samples/genericTypes69.py +45 -0
  694. package/src/tests/samples/genericTypes7.py +24 -0
  695. package/src/tests/samples/genericTypes70.py +31 -0
  696. package/src/tests/samples/genericTypes71.py +72 -0
  697. package/src/tests/samples/genericTypes72.py +24 -0
  698. package/src/tests/samples/genericTypes73.py +14 -0
  699. package/src/tests/samples/genericTypes74.py +23 -0
  700. package/src/tests/samples/genericTypes75.py +45 -0
  701. package/src/tests/samples/genericTypes76.py +107 -0
  702. package/src/tests/samples/genericTypes77.py +26 -0
  703. package/src/tests/samples/genericTypes78.py +23 -0
  704. package/src/tests/samples/genericTypes79.py +37 -0
  705. package/src/tests/samples/genericTypes8.py +26 -0
  706. package/src/tests/samples/genericTypes80.py +10 -0
  707. package/src/tests/samples/genericTypes81.py +32 -0
  708. package/src/tests/samples/genericTypes82.py +19 -0
  709. package/src/tests/samples/genericTypes83.py +28 -0
  710. package/src/tests/samples/genericTypes84.py +11 -0
  711. package/src/tests/samples/genericTypes85.py +20 -0
  712. package/src/tests/samples/genericTypes86.py +15 -0
  713. package/src/tests/samples/genericTypes87.py +21 -0
  714. package/src/tests/samples/genericTypes88.py +25 -0
  715. package/src/tests/samples/genericTypes89.py +17 -0
  716. package/src/tests/samples/genericTypes9.py +109 -0
  717. package/src/tests/samples/genericTypes90.py +12 -0
  718. package/src/tests/samples/genericTypes91.py +20 -0
  719. package/src/tests/samples/import1.py +7 -0
  720. package/src/tests/samples/import10.py +11 -0
  721. package/src/tests/samples/import11.py +11 -0
  722. package/src/tests/samples/import12.py +5 -0
  723. package/src/tests/samples/import13.py +6 -0
  724. package/src/tests/samples/import14.py +10 -0
  725. package/src/tests/samples/import2.py +27 -0
  726. package/src/tests/samples/import3.py +8 -0
  727. package/src/tests/samples/import4.py +15 -0
  728. package/src/tests/samples/import5.py +6 -0
  729. package/src/tests/samples/import6.py +17 -0
  730. package/src/tests/samples/import7.py +14 -0
  731. package/src/tests/samples/import8.py +7 -0
  732. package/src/tests/samples/import9.py +7 -0
  733. package/src/tests/samples/inconsistentConstructor1.py +23 -0
  734. package/src/tests/samples/inconsistentSpaceTab1.py +9 -0
  735. package/src/tests/samples/inconsistentSpaceTab2.py +10 -0
  736. package/src/tests/samples/index1.py +97 -0
  737. package/src/tests/samples/inferredTypes1.py +25 -0
  738. package/src/tests/samples/initVar1.py +23 -0
  739. package/src/tests/samples/initsubclass1.py +46 -0
  740. package/src/tests/samples/initsubclass2.py +16 -0
  741. package/src/tests/samples/isinstance1.py +57 -0
  742. package/src/tests/samples/isinstance10.py +15 -0
  743. package/src/tests/samples/isinstance2.py +28 -0
  744. package/src/tests/samples/isinstance3.py +59 -0
  745. package/src/tests/samples/isinstance4.py +57 -0
  746. package/src/tests/samples/isinstance5.py +8 -0
  747. package/src/tests/samples/isinstance6.py +19 -0
  748. package/src/tests/samples/isinstance7.py +42 -0
  749. package/src/tests/samples/isinstance9.py +26 -0
  750. package/src/tests/samples/kwargsUnpack1.py +116 -0
  751. package/src/tests/samples/lambda1.py +77 -0
  752. package/src/tests/samples/lambda2.py +26 -0
  753. package/src/tests/samples/lambda3.py +34 -0
  754. package/src/tests/samples/lambda4.py +70 -0
  755. package/src/tests/samples/lambda5.py +23 -0
  756. package/src/tests/samples/lambda6.py +15 -0
  757. package/src/tests/samples/lines1.py +14 -0
  758. package/src/tests/samples/list1.py +90 -0
  759. package/src/tests/samples/listComprehension1.py +45 -0
  760. package/src/tests/samples/listComprehension2.py +17 -0
  761. package/src/tests/samples/listComprehension3.py +12 -0
  762. package/src/tests/samples/listComprehension4.py +15 -0
  763. package/src/tests/samples/listComprehension5.py +20 -0
  764. package/src/tests/samples/listComprehension6.py +14 -0
  765. package/src/tests/samples/listComprehension7.py +15 -0
  766. package/src/tests/samples/listComprehension8.py +15 -0
  767. package/src/tests/samples/literalString1.py +43 -0
  768. package/src/tests/samples/literals1.py +48 -0
  769. package/src/tests/samples/literals2.py +30 -0
  770. package/src/tests/samples/literals3.py +40 -0
  771. package/src/tests/samples/literals4.py +10 -0
  772. package/src/tests/samples/literals5.py +28 -0
  773. package/src/tests/samples/literals6.py +90 -0
  774. package/src/tests/samples/literals7.py +38 -0
  775. package/src/tests/samples/loops1.py +45 -0
  776. package/src/tests/samples/loops10.py +11 -0
  777. package/src/tests/samples/loops11.py +22 -0
  778. package/src/tests/samples/loops12.py +18 -0
  779. package/src/tests/samples/loops13.py +13 -0
  780. package/src/tests/samples/loops14.py +22 -0
  781. package/src/tests/samples/loops15.py +22 -0
  782. package/src/tests/samples/loops16.py +322 -0
  783. package/src/tests/samples/loops17.py +9 -0
  784. package/src/tests/samples/loops18.py +23 -0
  785. package/src/tests/samples/loops19.py +14 -0
  786. package/src/tests/samples/loops2.py +25 -0
  787. package/src/tests/samples/loops20.py +41 -0
  788. package/src/tests/samples/loops21.py +18 -0
  789. package/src/tests/samples/loops22.py +21 -0
  790. package/src/tests/samples/loops23.py +16 -0
  791. package/src/tests/samples/loops24.py +12 -0
  792. package/src/tests/samples/loops25.py +45 -0
  793. package/src/tests/samples/loops26.py +12 -0
  794. package/src/tests/samples/loops3.py +11 -0
  795. package/src/tests/samples/loops4.py +17 -0
  796. package/src/tests/samples/loops5.py +14 -0
  797. package/src/tests/samples/loops6.py +25 -0
  798. package/src/tests/samples/loops7.py +14 -0
  799. package/src/tests/samples/loops8.py +17 -0
  800. package/src/tests/samples/loops9.py +25 -0
  801. package/src/tests/samples/match1.py +190 -0
  802. package/src/tests/samples/match10.py +67 -0
  803. package/src/tests/samples/match2.py +340 -0
  804. package/src/tests/samples/match3.py +364 -0
  805. package/src/tests/samples/match4.py +100 -0
  806. package/src/tests/samples/match5.py +70 -0
  807. package/src/tests/samples/match6.py +64 -0
  808. package/src/tests/samples/match7.py +65 -0
  809. package/src/tests/samples/match8.py +19 -0
  810. package/src/tests/samples/match9.py +53 -0
  811. package/src/tests/samples/maxParseDepth1.py +499 -0
  812. package/src/tests/samples/maxParseDepth2.py +16 -0
  813. package/src/tests/samples/memberAccess1.py +70 -0
  814. package/src/tests/samples/memberAccess10.py +65 -0
  815. package/src/tests/samples/memberAccess11.py +37 -0
  816. package/src/tests/samples/memberAccess12.py +36 -0
  817. package/src/tests/samples/memberAccess13.py +17 -0
  818. package/src/tests/samples/memberAccess14.py +63 -0
  819. package/src/tests/samples/memberAccess15.py +20 -0
  820. package/src/tests/samples/memberAccess16.py +12 -0
  821. package/src/tests/samples/memberAccess17.py +31 -0
  822. package/src/tests/samples/memberAccess18.py +31 -0
  823. package/src/tests/samples/memberAccess2.py +45 -0
  824. package/src/tests/samples/memberAccess3.py +64 -0
  825. package/src/tests/samples/memberAccess4.py +91 -0
  826. package/src/tests/samples/memberAccess5.py +26 -0
  827. package/src/tests/samples/memberAccess6.py +52 -0
  828. package/src/tests/samples/memberAccess7.py +36 -0
  829. package/src/tests/samples/memberAccess8.py +75 -0
  830. package/src/tests/samples/memberAccess9.py +11 -0
  831. package/src/tests/samples/metaclass1.py +7 -0
  832. package/src/tests/samples/metaclass2.py +35 -0
  833. package/src/tests/samples/metaclass3.py +43 -0
  834. package/src/tests/samples/metaclass4.py +26 -0
  835. package/src/tests/samples/metaclass5.py +33 -0
  836. package/src/tests/samples/metaclass6.py +15 -0
  837. package/src/tests/samples/metaclass7.py +38 -0
  838. package/src/tests/samples/metaclass8.py +20 -0
  839. package/src/tests/samples/metaclass9.py +77 -0
  840. package/src/tests/samples/methodOverride1.py +507 -0
  841. package/src/tests/samples/methodOverride2.py +74 -0
  842. package/src/tests/samples/methodOverride3.py +111 -0
  843. package/src/tests/samples/methodOverride4.py +36 -0
  844. package/src/tests/samples/missingSuper1.py +73 -0
  845. package/src/tests/samples/module1.py +13 -0
  846. package/src/tests/samples/module2.py +7 -0
  847. package/src/tests/samples/mro1.py +16 -0
  848. package/src/tests/samples/mro2.py +45 -0
  849. package/src/tests/samples/mro3.py +30 -0
  850. package/src/tests/samples/mro4.py +30 -0
  851. package/src/tests/samples/nameBindings1.py +48 -0
  852. package/src/tests/samples/nameBindings2.py +9 -0
  853. package/src/tests/samples/nameBindings3.py +36 -0
  854. package/src/tests/samples/nameBindings4.py +10 -0
  855. package/src/tests/samples/nameBindings5.py +53 -0
  856. package/src/tests/samples/namedTuples1.py +103 -0
  857. package/src/tests/samples/namedTuples2.py +56 -0
  858. package/src/tests/samples/namedTuples3.py +15 -0
  859. package/src/tests/samples/namedTuples4.py +24 -0
  860. package/src/tests/samples/namedTuples5.py +15 -0
  861. package/src/tests/samples/namedTuples6.py +31 -0
  862. package/src/tests/samples/namedTuples7.py +27 -0
  863. package/src/tests/samples/never1.py +12 -0
  864. package/src/tests/samples/never2.py +43 -0
  865. package/src/tests/samples/newType1.py +34 -0
  866. package/src/tests/samples/newType2.py +25 -0
  867. package/src/tests/samples/newType3.py +34 -0
  868. package/src/tests/samples/newType4.py +8 -0
  869. package/src/tests/samples/none1.py +29 -0
  870. package/src/tests/samples/none2.py +22 -0
  871. package/src/tests/samples/noreturn1.py +78 -0
  872. package/src/tests/samples/noreturn2.py +37 -0
  873. package/src/tests/samples/noreturn3.py +16 -0
  874. package/src/tests/samples/noreturn4.py +29 -0
  875. package/src/tests/samples/operators1.py +101 -0
  876. package/src/tests/samples/operators2.py +30 -0
  877. package/src/tests/samples/operators3.py +8 -0
  878. package/src/tests/samples/operators4.py +22 -0
  879. package/src/tests/samples/operators5.py +5 -0
  880. package/src/tests/samples/operators6.py +10 -0
  881. package/src/tests/samples/operators7.py +24 -0
  882. package/src/tests/samples/operators8.py +125 -0
  883. package/src/tests/samples/optional1.py +77 -0
  884. package/src/tests/samples/optional2.py +23 -0
  885. package/src/tests/samples/overload1.py +50 -0
  886. package/src/tests/samples/overload10.py +48 -0
  887. package/src/tests/samples/overload2.py +37 -0
  888. package/src/tests/samples/overload3.py +29 -0
  889. package/src/tests/samples/overload4.py +46 -0
  890. package/src/tests/samples/overload5.py +333 -0
  891. package/src/tests/samples/overload6.py +64 -0
  892. package/src/tests/samples/overload7.py +376 -0
  893. package/src/tests/samples/overload8.py +89 -0
  894. package/src/tests/samples/overload9.py +19 -0
  895. package/src/tests/samples/package1/__init__.py +3 -0
  896. package/src/tests/samples/package1/psyche/pysche.py +4 -0
  897. package/src/tests/samples/package1/psyche.py +4 -0
  898. package/src/tests/samples/package1/sub/__init__.py +5 -0
  899. package/src/tests/samples/package1/sub.py +4 -0
  900. package/src/tests/samples/package1/sub.pyi +2 -0
  901. package/src/tests/samples/paramInference1.py +27 -0
  902. package/src/tests/samples/paramNames1.py +39 -0
  903. package/src/tests/samples/paramSpec1.py +64 -0
  904. package/src/tests/samples/paramSpec10.py +57 -0
  905. package/src/tests/samples/paramSpec11.py +36 -0
  906. package/src/tests/samples/paramSpec12.py +88 -0
  907. package/src/tests/samples/paramSpec13.py +112 -0
  908. package/src/tests/samples/paramSpec14.py +30 -0
  909. package/src/tests/samples/paramSpec15.py +28 -0
  910. package/src/tests/samples/paramSpec16.py +28 -0
  911. package/src/tests/samples/paramSpec17.py +34 -0
  912. package/src/tests/samples/paramSpec18.py +43 -0
  913. package/src/tests/samples/paramSpec19.py +93 -0
  914. package/src/tests/samples/paramSpec2.py +29 -0
  915. package/src/tests/samples/paramSpec20.py +87 -0
  916. package/src/tests/samples/paramSpec21.py +58 -0
  917. package/src/tests/samples/paramSpec22.py +26 -0
  918. package/src/tests/samples/paramSpec23.py +21 -0
  919. package/src/tests/samples/paramSpec24.py +67 -0
  920. package/src/tests/samples/paramSpec25.py +36 -0
  921. package/src/tests/samples/paramSpec26.py +25 -0
  922. package/src/tests/samples/paramSpec27.py +57 -0
  923. package/src/tests/samples/paramSpec28.py +39 -0
  924. package/src/tests/samples/paramSpec29.py +37 -0
  925. package/src/tests/samples/paramSpec3.py +71 -0
  926. package/src/tests/samples/paramSpec30.py +49 -0
  927. package/src/tests/samples/paramSpec31.py +16 -0
  928. package/src/tests/samples/paramSpec32.py +57 -0
  929. package/src/tests/samples/paramSpec33.py +29 -0
  930. package/src/tests/samples/paramSpec34.py +29 -0
  931. package/src/tests/samples/paramSpec35.py +24 -0
  932. package/src/tests/samples/paramSpec36.py +43 -0
  933. package/src/tests/samples/paramSpec37.py +25 -0
  934. package/src/tests/samples/paramSpec4.py +109 -0
  935. package/src/tests/samples/paramSpec5.py +29 -0
  936. package/src/tests/samples/paramSpec6.py +17 -0
  937. package/src/tests/samples/paramSpec7.py +22 -0
  938. package/src/tests/samples/paramSpec8.py +45 -0
  939. package/src/tests/samples/paramSpec9.py +57 -0
  940. package/src/tests/samples/paramType1.py +71 -0
  941. package/src/tests/samples/parameters1.py +16 -0
  942. package/src/tests/samples/partial1.py +163 -0
  943. package/src/tests/samples/partial2.py +31 -0
  944. package/src/tests/samples/private1.py +55 -0
  945. package/src/tests/samples/private2.py +11 -0
  946. package/src/tests/samples/project1/pyrightconfig.json +5 -0
  947. package/src/tests/samples/project1/sample1.py +0 -0
  948. package/src/tests/samples/project1/subfolder1/sample2.py +0 -0
  949. package/src/tests/samples/project1/subfolder1/subfolder1-1/sample3.py +0 -0
  950. package/src/tests/samples/project2/pyrightconfig.json +3 -0
  951. package/src/tests/samples/project3/pyrightconfig.json +7 -0
  952. package/src/tests/samples/project4/presentfile.py +0 -0
  953. package/src/tests/samples/project4/pyrightconfig.json +11 -0
  954. package/src/tests/samples/project4/subfolder/presentfile2.py +0 -0
  955. package/src/tests/samples/project4/subfolder/presentfile3.py +0 -0
  956. package/src/tests/samples/project5/pyrightconfig.json +4 -0
  957. package/src/tests/samples/project6/app1.py +0 -0
  958. package/src/tests/samples/project6/app2.py +0 -0
  959. package/src/tests/samples/project6/projectA/foo/__init__.py +0 -0
  960. package/src/tests/samples/project6/projectA/foo/bar/__init__.py +0 -0
  961. package/src/tests/samples/project6/projectB/foo/__init__.py +0 -0
  962. package/src/tests/samples/project6/projectB/foo/baz/__init__.py +0 -0
  963. package/src/tests/samples/project_src/src/module1.py +0 -0
  964. package/src/tests/samples/project_src_is_pkg/src/__init__.py +0 -0
  965. package/src/tests/samples/project_src_is_pkg/src/module1.py +0 -0
  966. package/src/tests/samples/project_src_with_config_extra_paths/pyrightconfig.json +3 -0
  967. package/src/tests/samples/project_src_with_config_extra_paths/src/module1.py +0 -0
  968. package/src/tests/samples/project_src_with_config_no_extra_paths/pyrightconfig.json +2 -0
  969. package/src/tests/samples/project_src_with_config_no_extra_paths/src/module1.py +0 -0
  970. package/src/tests/samples/project_src_with_extra_paths/src/_vendored/vendored1.py +1 -0
  971. package/src/tests/samples/project_src_with_extra_paths/src/module1.py +2 -0
  972. package/src/tests/samples/project_with_pyproject_toml/pyproject.toml +7 -0
  973. package/src/tests/samples/project_with_venv_auto_detect_exclude/myvenv/Lib/site-packages/library1.py +0 -0
  974. package/src/tests/samples/project_with_venv_auto_detect_exclude/myvenv/pyvenv.cfg +0 -0
  975. package/src/tests/samples/project_with_venv_auto_detect_exclude/pyrightconfig.json +2 -0
  976. package/src/tests/samples/project_with_venv_auto_detect_exclude/sample1.py +0 -0
  977. package/src/tests/samples/project_with_venv_auto_detect_exclude/subfolder1/sample2.py +0 -0
  978. package/src/tests/samples/project_with_venv_auto_detect_exclude/subfolder1/subfolder1-1/sample3.py +0 -0
  979. package/src/tests/samples/project_with_venv_auto_detect_include/excluded/excluded1.py +0 -0
  980. package/src/tests/samples/project_with_venv_auto_detect_include/myvenv/Lib/site-packages/library1.py +0 -0
  981. package/src/tests/samples/project_with_venv_auto_detect_include/myvenv/pyvenv.cfg +0 -0
  982. package/src/tests/samples/project_with_venv_auto_detect_include/pyrightconfig.json +3 -0
  983. package/src/tests/samples/project_with_venv_auto_detect_include/sample1.py +0 -0
  984. package/src/tests/samples/project_with_venv_auto_detect_include/subfolder1/sample2.py +0 -0
  985. package/src/tests/samples/project_with_venv_auto_detect_include/subfolder1/subfolder1-1/sample3.py +0 -0
  986. package/src/tests/samples/properties1.py +64 -0
  987. package/src/tests/samples/properties10.py +19 -0
  988. package/src/tests/samples/properties11.py +45 -0
  989. package/src/tests/samples/properties12.py +23 -0
  990. package/src/tests/samples/properties13.py +15 -0
  991. package/src/tests/samples/properties2.py +34 -0
  992. package/src/tests/samples/properties3.py +93 -0
  993. package/src/tests/samples/properties4.py +17 -0
  994. package/src/tests/samples/properties5.py +29 -0
  995. package/src/tests/samples/properties6.py +35 -0
  996. package/src/tests/samples/properties7.py +30 -0
  997. package/src/tests/samples/properties8.py +26 -0
  998. package/src/tests/samples/properties9.py +24 -0
  999. package/src/tests/samples/protocol1.py +132 -0
  1000. package/src/tests/samples/protocol10.py +36 -0
  1001. package/src/tests/samples/protocol11.py +32 -0
  1002. package/src/tests/samples/protocol12.py +12 -0
  1003. package/src/tests/samples/protocol13.py +26 -0
  1004. package/src/tests/samples/protocol14.py +27 -0
  1005. package/src/tests/samples/protocol15.py +27 -0
  1006. package/src/tests/samples/protocol16.py +22 -0
  1007. package/src/tests/samples/protocol17.py +94 -0
  1008. package/src/tests/samples/protocol18.py +27 -0
  1009. package/src/tests/samples/protocol19.py +39 -0
  1010. package/src/tests/samples/protocol2.py +38 -0
  1011. package/src/tests/samples/protocol20.py +22 -0
  1012. package/src/tests/samples/protocol21.py +26 -0
  1013. package/src/tests/samples/protocol22.py +43 -0
  1014. package/src/tests/samples/protocol23.py +41 -0
  1015. package/src/tests/samples/protocol24.py +74 -0
  1016. package/src/tests/samples/protocol25.py +31 -0
  1017. package/src/tests/samples/protocol26.py +40 -0
  1018. package/src/tests/samples/protocol27.py +76 -0
  1019. package/src/tests/samples/protocol28.py +32 -0
  1020. package/src/tests/samples/protocol29.py +26 -0
  1021. package/src/tests/samples/protocol3.py +110 -0
  1022. package/src/tests/samples/protocol30.py +30 -0
  1023. package/src/tests/samples/protocol31.py +22 -0
  1024. package/src/tests/samples/protocol32.py +67 -0
  1025. package/src/tests/samples/protocol33.py +39 -0
  1026. package/src/tests/samples/protocol34.py +21 -0
  1027. package/src/tests/samples/protocol4.py +22 -0
  1028. package/src/tests/samples/protocol5.py +25 -0
  1029. package/src/tests/samples/protocol6.py +65 -0
  1030. package/src/tests/samples/protocol7.py +24 -0
  1031. package/src/tests/samples/protocol8.py +34 -0
  1032. package/src/tests/samples/protocol9.py +35 -0
  1033. package/src/tests/samples/protocolModule1.py +15 -0
  1034. package/src/tests/samples/protocolModule2.py +83 -0
  1035. package/src/tests/samples/protocolModule3.py +14 -0
  1036. package/src/tests/samples/protocolModule4.py +26 -0
  1037. package/src/tests/samples/pseudoGeneric1.py +20 -0
  1038. package/src/tests/samples/pyrightIgnore1.py +11 -0
  1039. package/src/tests/samples/pyrightIgnore2.py +24 -0
  1040. package/src/tests/samples/python2.py +31 -0
  1041. package/src/tests/samples/recursiveTypeAlias1.py +79 -0
  1042. package/src/tests/samples/recursiveTypeAlias2.py +32 -0
  1043. package/src/tests/samples/recursiveTypeAlias3.py +26 -0
  1044. package/src/tests/samples/recursiveTypeAlias4.py +55 -0
  1045. package/src/tests/samples/recursiveTypeAlias5.pyi +10 -0
  1046. package/src/tests/samples/recursiveTypeAlias6.py +25 -0
  1047. package/src/tests/samples/recursiveTypeAlias7.py +15 -0
  1048. package/src/tests/samples/recursiveTypeAlias8.py +38 -0
  1049. package/src/tests/samples/recursiveTypeAlias9.py +22 -0
  1050. package/src/tests/samples/required1.py +47 -0
  1051. package/src/tests/samples/required2.py +56 -0
  1052. package/src/tests/samples/required3.py +22 -0
  1053. package/src/tests/samples/returnTypes1.py +26 -0
  1054. package/src/tests/samples/revealedType1.py +30 -0
  1055. package/src/tests/samples/sample1.py +6 -0
  1056. package/src/tests/samples/self1.py +93 -0
  1057. package/src/tests/samples/self2.py +195 -0
  1058. package/src/tests/samples/self3.py +9 -0
  1059. package/src/tests/samples/self4.py +18 -0
  1060. package/src/tests/samples/self5.py +26 -0
  1061. package/src/tests/samples/setComprehension1.py +34 -0
  1062. package/src/tests/samples/slots1.py +63 -0
  1063. package/src/tests/samples/slots2.py +38 -0
  1064. package/src/tests/samples/slots3.py +35 -0
  1065. package/src/tests/samples/specialization1.py +78 -0
  1066. package/src/tests/samples/specialization2.py +51 -0
  1067. package/src/tests/samples/staticExpressions1.py +52 -0
  1068. package/src/tests/samples/strings1.py +22 -0
  1069. package/src/tests/samples/subscript1.py +74 -0
  1070. package/src/tests/samples/subscript2.py +55 -0
  1071. package/src/tests/samples/subscript3.py +135 -0
  1072. package/src/tests/samples/suiteExpectedColon1.py +4 -0
  1073. package/src/tests/samples/suiteExpectedColon2.py +2 -0
  1074. package/src/tests/samples/suiteExpectedColon3.py +2 -0
  1075. package/src/tests/samples/super1.py +57 -0
  1076. package/src/tests/samples/super2.py +29 -0
  1077. package/src/tests/samples/super3.py +10 -0
  1078. package/src/tests/samples/super4.py +24 -0
  1079. package/src/tests/samples/super5.py +28 -0
  1080. package/src/tests/samples/super6.py +17 -0
  1081. package/src/tests/samples/super7.py +52 -0
  1082. package/src/tests/samples/super8.py +7 -0
  1083. package/src/tests/samples/super9.py +36 -0
  1084. package/src/tests/samples/test_file1.py +0 -0
  1085. package/src/tests/samples/threePartVersion1.py +33 -0
  1086. package/src/tests/samples/totalOrdering1.py +52 -0
  1087. package/src/tests/samples/tryExcept1.py +39 -0
  1088. package/src/tests/samples/tryExcept10.py +12 -0
  1089. package/src/tests/samples/tryExcept2.py +16 -0
  1090. package/src/tests/samples/tryExcept3.py +12 -0
  1091. package/src/tests/samples/tryExcept4.py +28 -0
  1092. package/src/tests/samples/tryExcept5.py +30 -0
  1093. package/src/tests/samples/tryExcept6.py +27 -0
  1094. package/src/tests/samples/tryExcept7.py +17 -0
  1095. package/src/tests/samples/tryExcept8.py +69 -0
  1096. package/src/tests/samples/tryExcept9.py +42 -0
  1097. package/src/tests/samples/tupleUnpack1.py +63 -0
  1098. package/src/tests/samples/tupleUnpack2.py +56 -0
  1099. package/src/tests/samples/tupleUnpack3.py +28 -0
  1100. package/src/tests/samples/tuples1.py +185 -0
  1101. package/src/tests/samples/tuples10.py +30 -0
  1102. package/src/tests/samples/tuples11.py +21 -0
  1103. package/src/tests/samples/tuples12.py +31 -0
  1104. package/src/tests/samples/tuples13.py +26 -0
  1105. package/src/tests/samples/tuples15.py +16 -0
  1106. package/src/tests/samples/tuples16.py +14 -0
  1107. package/src/tests/samples/tuples2.py +23 -0
  1108. package/src/tests/samples/tuples3.py +15 -0
  1109. package/src/tests/samples/tuples4.py +21 -0
  1110. package/src/tests/samples/tuples5.py +16 -0
  1111. package/src/tests/samples/tuples6.py +56 -0
  1112. package/src/tests/samples/tuples7.py +61 -0
  1113. package/src/tests/samples/tuples8.py +98 -0
  1114. package/src/tests/samples/tuples9.py +17 -0
  1115. package/src/tests/samples/typeAlias1.py +37 -0
  1116. package/src/tests/samples/typeAlias10.py +45 -0
  1117. package/src/tests/samples/typeAlias11.py +37 -0
  1118. package/src/tests/samples/typeAlias12.py +20 -0
  1119. package/src/tests/samples/typeAlias13.py +53 -0
  1120. package/src/tests/samples/typeAlias14.py +29 -0
  1121. package/src/tests/samples/typeAlias15.py +24 -0
  1122. package/src/tests/samples/typeAlias16.py +24 -0
  1123. package/src/tests/samples/typeAlias2.py +26 -0
  1124. package/src/tests/samples/typeAlias3.py +35 -0
  1125. package/src/tests/samples/typeAlias4.py +68 -0
  1126. package/src/tests/samples/typeAlias5.py +56 -0
  1127. package/src/tests/samples/typeAlias6.py +51 -0
  1128. package/src/tests/samples/typeAlias7.py +50 -0
  1129. package/src/tests/samples/typeAlias8.py +24 -0
  1130. package/src/tests/samples/typeAlias9.py +35 -0
  1131. package/src/tests/samples/typeGuard1.py +68 -0
  1132. package/src/tests/samples/typeGuard2.py +58 -0
  1133. package/src/tests/samples/typeGuard3.py +98 -0
  1134. package/src/tests/samples/typeIgnore1.py +10 -0
  1135. package/src/tests/samples/typeIgnore2.py +20 -0
  1136. package/src/tests/samples/typeIgnore3.py +22 -0
  1137. package/src/tests/samples/typeIgnore4.py +10 -0
  1138. package/src/tests/samples/typeIgnore5.py +7 -0
  1139. package/src/tests/samples/typeNarrowing1.py +56 -0
  1140. package/src/tests/samples/typeNarrowing2.py +14 -0
  1141. package/src/tests/samples/typeNarrowing3.py +47 -0
  1142. package/src/tests/samples/typeNarrowing4.py +24 -0
  1143. package/src/tests/samples/typeNarrowing5.py +19 -0
  1144. package/src/tests/samples/typeNarrowing6.py +27 -0
  1145. package/src/tests/samples/typeNarrowing7.py +71 -0
  1146. package/src/tests/samples/typeNarrowingAssert1.py +24 -0
  1147. package/src/tests/samples/typeNarrowingCallable1.py +78 -0
  1148. package/src/tests/samples/typeNarrowingEnum1.py +58 -0
  1149. package/src/tests/samples/typeNarrowingEnum2.py +71 -0
  1150. package/src/tests/samples/typeNarrowingFalsy1.py +50 -0
  1151. package/src/tests/samples/typeNarrowingIn1.py +97 -0
  1152. package/src/tests/samples/typeNarrowingIsNone1.py +56 -0
  1153. package/src/tests/samples/typeNarrowingIsNone2.py +8 -0
  1154. package/src/tests/samples/typeNarrowingIsNoneTuple1.py +42 -0
  1155. package/src/tests/samples/typeNarrowingIsNoneTuple2.py +35 -0
  1156. package/src/tests/samples/typeNarrowingIsinstance1.py +162 -0
  1157. package/src/tests/samples/typeNarrowingIsinstance2.py +14 -0
  1158. package/src/tests/samples/typeNarrowingIsinstance3.py +78 -0
  1159. package/src/tests/samples/typeNarrowingIsinstance4.py +50 -0
  1160. package/src/tests/samples/typeNarrowingIsinstance5.py +39 -0
  1161. package/src/tests/samples/typeNarrowingIsinstance6.py +77 -0
  1162. package/src/tests/samples/typeNarrowingIsinstance7.py +19 -0
  1163. package/src/tests/samples/typeNarrowingIsinstance8.py +19 -0
  1164. package/src/tests/samples/typeNarrowingLiteral1.py +35 -0
  1165. package/src/tests/samples/typeNarrowingLiteral2.py +63 -0
  1166. package/src/tests/samples/typeNarrowingLiteralMember1.py +140 -0
  1167. package/src/tests/samples/typeNarrowingLocalConst1.py +136 -0
  1168. package/src/tests/samples/typeNarrowingNoneMember1.py +102 -0
  1169. package/src/tests/samples/typeNarrowingTuple1.py +23 -0
  1170. package/src/tests/samples/typeNarrowingTupleLength1.py +47 -0
  1171. package/src/tests/samples/typeNarrowingTypeIs1.py +95 -0
  1172. package/src/tests/samples/typeNarrowingTypedDict1.py +99 -0
  1173. package/src/tests/samples/typeNarrowingTypedDict2.py +35 -0
  1174. package/src/tests/samples/typeNarrowingTypedDict3.py +45 -0
  1175. package/src/tests/samples/typePromotions1.py +30 -0
  1176. package/src/tests/samples/typeVar1.py +23 -0
  1177. package/src/tests/samples/typeVar10.py +21 -0
  1178. package/src/tests/samples/typeVar11.py +12 -0
  1179. package/src/tests/samples/typeVar2.py +26 -0
  1180. package/src/tests/samples/typeVar3.py +73 -0
  1181. package/src/tests/samples/typeVar4.py +45 -0
  1182. package/src/tests/samples/typeVar5.py +75 -0
  1183. package/src/tests/samples/typeVar6.py +115 -0
  1184. package/src/tests/samples/typeVar7.py +171 -0
  1185. package/src/tests/samples/typeVar8.py +30 -0
  1186. package/src/tests/samples/typeVar9.py +118 -0
  1187. package/src/tests/samples/typedDict1.py +63 -0
  1188. package/src/tests/samples/typedDict10.py +34 -0
  1189. package/src/tests/samples/typedDict11.py +12 -0
  1190. package/src/tests/samples/typedDict12.py +103 -0
  1191. package/src/tests/samples/typedDict13.py +14 -0
  1192. package/src/tests/samples/typedDict14.py +25 -0
  1193. package/src/tests/samples/typedDict15.py +53 -0
  1194. package/src/tests/samples/typedDict16.py +103 -0
  1195. package/src/tests/samples/typedDict17.py +24 -0
  1196. package/src/tests/samples/typedDict18.py +43 -0
  1197. package/src/tests/samples/typedDict2.py +56 -0
  1198. package/src/tests/samples/typedDict3.py +40 -0
  1199. package/src/tests/samples/typedDict4.py +68 -0
  1200. package/src/tests/samples/typedDict5.py +50 -0
  1201. package/src/tests/samples/typedDict6.py +77 -0
  1202. package/src/tests/samples/typedDict7.py +22 -0
  1203. package/src/tests/samples/typedDict8.py +19 -0
  1204. package/src/tests/samples/typedDict9.py +22 -0
  1205. package/src/tests/samples/unbound1.py +18 -0
  1206. package/src/tests/samples/unbound2.py +12 -0
  1207. package/src/tests/samples/unbound3.py +13 -0
  1208. package/src/tests/samples/unbound4.py +25 -0
  1209. package/src/tests/samples/unicode1.py +39 -0
  1210. package/src/tests/samples/uninitializedVariable1.py +27 -0
  1211. package/src/tests/samples/unions1.py +79 -0
  1212. package/src/tests/samples/unions2.py +15 -0
  1213. package/src/tests/samples/unions3.py +48 -0
  1214. package/src/tests/samples/unions4.py +20 -0
  1215. package/src/tests/samples/unions5.py +43 -0
  1216. package/src/tests/samples/unnecessaryCast1.py +13 -0
  1217. package/src/tests/samples/unnecessaryIsInstance1.py +91 -0
  1218. package/src/tests/samples/unnecessaryIsSubclass1.py +22 -0
  1219. package/src/tests/samples/unpack1.py +56 -0
  1220. package/src/tests/samples/unpack2.py +27 -0
  1221. package/src/tests/samples/unpack3.py +16 -0
  1222. package/src/tests/samples/unpack4.py +15 -0
  1223. package/src/tests/samples/unreachable1.py +112 -0
  1224. package/src/tests/samples/unusedCallResult1.py +67 -0
  1225. package/src/tests/samples/unusedCoroutine1.py +23 -0
  1226. package/src/tests/samples/unusedExpression1.py +34 -0
  1227. package/src/tests/samples/unusedVariable1.py +21 -0
  1228. package/src/tests/samples/variadicTypeVar1.py +51 -0
  1229. package/src/tests/samples/variadicTypeVar10.py +65 -0
  1230. package/src/tests/samples/variadicTypeVar11.py +72 -0
  1231. package/src/tests/samples/variadicTypeVar12.py +26 -0
  1232. package/src/tests/samples/variadicTypeVar13.py +35 -0
  1233. package/src/tests/samples/variadicTypeVar14.py +72 -0
  1234. package/src/tests/samples/variadicTypeVar2.py +63 -0
  1235. package/src/tests/samples/variadicTypeVar3.py +86 -0
  1236. package/src/tests/samples/variadicTypeVar4.py +71 -0
  1237. package/src/tests/samples/variadicTypeVar5.py +131 -0
  1238. package/src/tests/samples/variadicTypeVar6.py +78 -0
  1239. package/src/tests/samples/variadicTypeVar7.py +59 -0
  1240. package/src/tests/samples/variadicTypeVar8.py +116 -0
  1241. package/src/tests/samples/variadicTypeVar9.py +38 -0
  1242. package/src/tests/samples/with1.py +117 -0
  1243. package/src/tests/samples/with2.py +63 -0
  1244. package/src/tests/samples/with3.py +66 -0
  1245. package/src/tests/samples/with4.py +24 -0
  1246. package/src/tests/samples/with5.py +33 -0
  1247. package/src/tests/samples/zipfs/bad.egg +1 -0
  1248. package/src/tests/samples/zipfs/bad.zip +1 -0
  1249. package/src/tests/samples/zipfs/basic.egg +0 -0
  1250. package/src/tests/samples/zipfs/basic.zip +0 -0
  1251. package/src/tests/samples/zipfs/corrupt.egg +0 -0
  1252. package/src/tests/samples/zipfs/corrupt.zip +0 -0
  1253. package/src/tests/sourceFile.test.ts +25 -0
  1254. package/src/tests/stringUtils.test.ts +63 -0
  1255. package/src/tests/symbolNameUtils.test.ts +77 -0
  1256. package/src/tests/testState.test.ts +579 -0
  1257. package/src/tests/testUtils.ts +251 -0
  1258. package/src/tests/tokenizer.test.ts +1534 -0
  1259. package/src/tests/typeEvaluator1.test.ts +1372 -0
  1260. package/src/tests/typeEvaluator2.test.ts +1253 -0
  1261. package/src/tests/typeEvaluator3.test.ts +1279 -0
  1262. package/src/tests/typeEvaluator4.test.ts +1178 -0
  1263. package/src/tests/updateSymbolReference.test.ts +1107 -0
  1264. package/src/tests/zipfs.test.ts +116 -0
  1265. package/src/workspaceMap.ts +76 -0
@@ -0,0 +1,1356 @@
1
+ /*
2
+ * patternMatching.ts
3
+ * Copyright (c) Microsoft Corporation.
4
+ * Licensed under the MIT license.
5
+ * Author: Eric Traut
6
+ *
7
+ * Type evaluation logic for evaluating and narrowing types
8
+ * related to "match" and "case" statements as documented in
9
+ * PEP 634.
10
+ */
11
+
12
+ import { appendArray } from '../common/collectionUtils';
13
+ import { assert } from '../common/debug';
14
+ import { DiagnosticRule } from '../common/diagnosticRules';
15
+ import { Localizer } from '../localization/localize';
16
+ import {
17
+ ExpressionNode,
18
+ ParseNode,
19
+ ParseNodeType,
20
+ PatternAsNode,
21
+ PatternAtomNode,
22
+ PatternClassArgumentNode,
23
+ PatternClassNode,
24
+ PatternLiteralNode,
25
+ PatternMappingNode,
26
+ PatternSequenceNode,
27
+ PatternValueNode,
28
+ } from '../parser/parseNodes';
29
+ import { getFileInfo } from './analyzerNodeInfo';
30
+ import { populateTypeVarContextBasedOnExpectedType } from './constraintSolver';
31
+ import { getTypedDictMembersForClass } from './typedDicts';
32
+ import { EvaluatorFlags, TypeEvaluator } from './typeEvaluatorTypes';
33
+ import { enumerateLiteralsForType } from './typeGuards';
34
+ import {
35
+ AnyType,
36
+ ClassType,
37
+ combineTypes,
38
+ isAnyOrUnknown,
39
+ isClass,
40
+ isClassInstance,
41
+ isInstantiableClass,
42
+ isNever,
43
+ isNoneInstance,
44
+ isSameWithoutLiteralValue,
45
+ isUnknown,
46
+ NeverType,
47
+ Type,
48
+ TypeBase,
49
+ UnknownType,
50
+ } from './types';
51
+ import {
52
+ addConditionToType,
53
+ applySolvedTypeVars,
54
+ convertToInstance,
55
+ doForEachSubtype,
56
+ getTypeCondition,
57
+ getTypeVarScopeId,
58
+ isLiteralType,
59
+ isTupleClass,
60
+ isUnboundedTupleClass,
61
+ lookUpClassMember,
62
+ mapSubtypes,
63
+ partiallySpecializeType,
64
+ specializeClassType,
65
+ specializeTupleClass,
66
+ stripLiteralValue,
67
+ } from './typeUtils';
68
+ import { TypeVarContext } from './typeVarContext';
69
+
70
+ // PEP 634 indicates that several built-in classes are handled differently
71
+ // when used with class pattern matching.
72
+ const classPatternSpecialCases = [
73
+ 'builtins.bool',
74
+ 'builtins.bytearray',
75
+ 'builtins.bytes',
76
+ 'builtins.dict',
77
+ 'builtins.float',
78
+ 'builtins.frozenset',
79
+ 'builtins.int',
80
+ 'builtins.list',
81
+ 'builtins.set',
82
+ 'builtins.str',
83
+ 'builtins.tuple',
84
+ ];
85
+
86
+ interface SequencePatternInfo {
87
+ subtype: Type;
88
+ definiteNoMatch: boolean;
89
+ entryTypes: Type[];
90
+ isIndeterminateLength?: boolean;
91
+ isTuple?: boolean;
92
+ isObject?: boolean;
93
+ }
94
+
95
+ interface MappingPatternInfo {
96
+ subtype: Type;
97
+ typedDict?: ClassType;
98
+ dictTypeArgs?: {
99
+ key: Type;
100
+ value: Type;
101
+ };
102
+ }
103
+
104
+ export function narrowTypeBasedOnPattern(
105
+ evaluator: TypeEvaluator,
106
+ type: Type,
107
+ pattern: PatternAtomNode,
108
+ isPositiveTest: boolean
109
+ ): Type {
110
+ switch (pattern.nodeType) {
111
+ case ParseNodeType.PatternSequence: {
112
+ return narrowTypeBasedOnSequencePattern(evaluator, type, pattern, isPositiveTest);
113
+ }
114
+
115
+ case ParseNodeType.PatternLiteral: {
116
+ return narrowTypeBasedOnLiteralPattern(evaluator, type, pattern, isPositiveTest);
117
+ }
118
+
119
+ case ParseNodeType.PatternClass: {
120
+ return narrowTypeBasedOnClassPattern(evaluator, type, pattern, isPositiveTest);
121
+ }
122
+
123
+ case ParseNodeType.PatternAs: {
124
+ return narrowTypeBasedOnAsPattern(evaluator, type, pattern, isPositiveTest);
125
+ }
126
+
127
+ case ParseNodeType.PatternMapping: {
128
+ return narrowTypeBasedOnMappingPattern(evaluator, type, pattern, isPositiveTest);
129
+ }
130
+
131
+ case ParseNodeType.PatternValue: {
132
+ return narrowTypeBasedOnValuePattern(evaluator, type, pattern, isPositiveTest);
133
+ }
134
+
135
+ case ParseNodeType.PatternCapture: {
136
+ // A capture captures everything, so nothing remains in the negative case.
137
+ return isPositiveTest ? type : NeverType.createNever();
138
+ }
139
+
140
+ case ParseNodeType.Error: {
141
+ return type;
142
+ }
143
+ }
144
+ }
145
+
146
+ function narrowTypeBasedOnSequencePattern(
147
+ evaluator: TypeEvaluator,
148
+ type: Type,
149
+ pattern: PatternSequenceNode,
150
+ isPositiveTest: boolean
151
+ ): Type {
152
+ let sequenceInfo = getSequencePatternInfo(evaluator, type, pattern.entries.length, pattern.starEntryIndex);
153
+
154
+ // Further narrow based on pattern entry types.
155
+ sequenceInfo = sequenceInfo.filter((entry) => {
156
+ if (entry.definiteNoMatch) {
157
+ if (isPositiveTest) {
158
+ return false;
159
+ } else {
160
+ return true;
161
+ }
162
+ }
163
+
164
+ let isPlausibleMatch = true;
165
+ let isDefiniteMatch = true;
166
+ const narrowedEntryTypes: Type[] = [];
167
+ let canNarrowTuple = entry.isTuple;
168
+
169
+ // If the subject has an indeterminate length but the pattern does not accept
170
+ // an arbitrary number of entries or accepts at least one non-star entry,
171
+ // we can't prove that it's a definite match.
172
+ if (entry.isIndeterminateLength) {
173
+ if (pattern.entries.length !== 1 || pattern.starEntryIndex !== 0) {
174
+ isDefiniteMatch = false;
175
+ }
176
+ }
177
+
178
+ pattern.entries.forEach((sequenceEntry, index) => {
179
+ const entryType = getTypeOfPatternSequenceEntry(
180
+ evaluator,
181
+ pattern,
182
+ entry,
183
+ index,
184
+ pattern.entries.length,
185
+ pattern.starEntryIndex,
186
+ /* unpackStarEntry */ true,
187
+ /* isSubjectObject */ false
188
+ );
189
+
190
+ const narrowedEntryType = narrowTypeBasedOnPattern(evaluator, entryType, sequenceEntry, isPositiveTest);
191
+
192
+ if (isPositiveTest) {
193
+ if (index === pattern.starEntryIndex) {
194
+ if (
195
+ isClassInstance(narrowedEntryType) &&
196
+ narrowedEntryType.tupleTypeArguments &&
197
+ !isUnboundedTupleClass(narrowedEntryType) &&
198
+ narrowedEntryType.tupleTypeArguments
199
+ ) {
200
+ appendArray(
201
+ narrowedEntryTypes,
202
+ narrowedEntryType.tupleTypeArguments.map((t) => t.type)
203
+ );
204
+ } else {
205
+ narrowedEntryTypes.push(narrowedEntryType);
206
+ canNarrowTuple = false;
207
+ }
208
+ } else {
209
+ narrowedEntryTypes.push(narrowedEntryType);
210
+
211
+ if (isNever(narrowedEntryType)) {
212
+ isPlausibleMatch = false;
213
+ }
214
+ }
215
+ } else {
216
+ if (!isNever(narrowedEntryType) || isAnyOrUnknown(entryType)) {
217
+ isDefiniteMatch = false;
218
+ }
219
+ }
220
+ });
221
+
222
+ if (!isPositiveTest) {
223
+ return !isDefiniteMatch;
224
+ }
225
+
226
+ if (isPlausibleMatch) {
227
+ // If this is a tuple, we can narrow it to a specific tuple type.
228
+ // Other sequences cannot be narrowed because we don't know if they
229
+ // are immutable (covariant).
230
+ if (canNarrowTuple) {
231
+ const tupleClassType = evaluator.getBuiltInType(pattern, 'tuple');
232
+ if (tupleClassType && isInstantiableClass(tupleClassType)) {
233
+ entry.subtype = ClassType.cloneAsInstance(
234
+ specializeTupleClass(
235
+ tupleClassType,
236
+ narrowedEntryTypes.map((t) => {
237
+ return { type: t, isUnbounded: false };
238
+ })
239
+ )
240
+ );
241
+ }
242
+ }
243
+
244
+ // If this is an object, we can narrow it to a specific Sequence type.
245
+ if (entry.isObject) {
246
+ const sequenceType = evaluator.getTypingType(pattern, 'Sequence');
247
+ if (sequenceType && isInstantiableClass(sequenceType)) {
248
+ entry.subtype = ClassType.cloneAsInstance(
249
+ ClassType.cloneForSpecialization(
250
+ sequenceType,
251
+ [stripLiteralValue(combineTypes(narrowedEntryTypes))],
252
+ /* isTypeArgumentExplicit */ true
253
+ )
254
+ );
255
+ }
256
+ }
257
+ }
258
+
259
+ return isPlausibleMatch;
260
+ });
261
+
262
+ return combineTypes(sequenceInfo.map((entry) => entry.subtype));
263
+ }
264
+
265
+ function narrowTypeBasedOnAsPattern(
266
+ evaluator: TypeEvaluator,
267
+ type: Type,
268
+ pattern: PatternAsNode,
269
+ isPositiveTest: boolean
270
+ ): Type {
271
+ let remainingType = type;
272
+
273
+ if (!isPositiveTest) {
274
+ pattern.orPatterns.forEach((subpattern) => {
275
+ remainingType = narrowTypeBasedOnPattern(evaluator, remainingType, subpattern, /* isPositiveTest */ false);
276
+ });
277
+ return remainingType;
278
+ }
279
+
280
+ const narrowedTypes = pattern.orPatterns.map((subpattern) => {
281
+ const narrowedSubtype = narrowTypeBasedOnPattern(
282
+ evaluator,
283
+ remainingType,
284
+ subpattern,
285
+ /* isPositiveTest */ true
286
+ );
287
+ remainingType = narrowTypeBasedOnPattern(evaluator, remainingType, subpattern, /* isPositiveTest */ false);
288
+ return narrowedSubtype;
289
+ });
290
+ return combineTypes(narrowedTypes);
291
+ }
292
+
293
+ function narrowTypeBasedOnMappingPattern(
294
+ evaluator: TypeEvaluator,
295
+ type: Type,
296
+ pattern: PatternMappingNode,
297
+ isPositiveTest: boolean
298
+ ): Type {
299
+ if (!isPositiveTest) {
300
+ // Never narrow in negative case.
301
+ return type;
302
+ }
303
+
304
+ let mappingInfo = getMappingPatternInfo(evaluator, type);
305
+
306
+ // Further narrow based on pattern entry types.
307
+ mappingInfo = mappingInfo.filter((mappingSubtypeInfo) => {
308
+ let isPlausibleMatch = true;
309
+ pattern.entries.forEach((mappingEntry) => {
310
+ if (mappingSubtypeInfo.typedDict) {
311
+ if (mappingEntry.nodeType === ParseNodeType.PatternMappingKeyEntry) {
312
+ const narrowedKeyType = narrowTypeBasedOnPattern(
313
+ evaluator,
314
+ evaluator.getBuiltInObject(pattern, 'str'),
315
+ mappingEntry.keyPattern,
316
+ isPositiveTest
317
+ );
318
+
319
+ if (isNever(narrowedKeyType)) {
320
+ isPlausibleMatch = false;
321
+ }
322
+
323
+ const valueType = mapSubtypes(narrowedKeyType, (keySubtype) => {
324
+ if (isAnyOrUnknown(keySubtype)) {
325
+ return keySubtype;
326
+ }
327
+
328
+ if (isClassInstance(keySubtype) && ClassType.isBuiltIn(keySubtype, 'str')) {
329
+ if (!isLiteralType(keySubtype)) {
330
+ return UnknownType.create();
331
+ }
332
+
333
+ const tdEntries = getTypedDictMembersForClass(evaluator, mappingSubtypeInfo.typedDict!);
334
+ const valueEntry = tdEntries.get(keySubtype.literalValue as string);
335
+ if (valueEntry) {
336
+ const narrowedValueType = narrowTypeBasedOnPattern(
337
+ evaluator,
338
+ valueEntry.valueType,
339
+ mappingEntry.valuePattern,
340
+ /* isPositiveTest */ true
341
+ );
342
+ if (!isNever(narrowedValueType)) {
343
+ return narrowedValueType;
344
+ }
345
+ }
346
+ }
347
+
348
+ return undefined;
349
+ });
350
+
351
+ if (isNever(valueType)) {
352
+ isPlausibleMatch = false;
353
+ }
354
+ }
355
+ } else if (mappingSubtypeInfo.dictTypeArgs) {
356
+ if (mappingEntry.nodeType === ParseNodeType.PatternMappingKeyEntry) {
357
+ const narrowedKeyType = narrowTypeBasedOnPattern(
358
+ evaluator,
359
+ mappingSubtypeInfo.dictTypeArgs.key,
360
+ mappingEntry.keyPattern,
361
+ isPositiveTest
362
+ );
363
+ const narrowedValueType = narrowTypeBasedOnPattern(
364
+ evaluator,
365
+ mappingSubtypeInfo.dictTypeArgs.value,
366
+ mappingEntry.valuePattern,
367
+ isPositiveTest
368
+ );
369
+ if (isNever(narrowedKeyType) || isNever(narrowedValueType)) {
370
+ isPlausibleMatch = false;
371
+ }
372
+ }
373
+ }
374
+ });
375
+
376
+ return isPlausibleMatch;
377
+ });
378
+
379
+ return combineTypes(mappingInfo.map((entry) => entry.subtype));
380
+ }
381
+
382
+ // Looks up the "__match_args__" class member to determine the names of
383
+ // the attributes used for class pattern matching.
384
+ function getPositionalMatchArgNames(evaluator: TypeEvaluator, type: ClassType): string[] {
385
+ const matchArgsMemberInfo = lookUpClassMember(type, '__match_args__');
386
+ if (matchArgsMemberInfo) {
387
+ const matchArgsType = evaluator.getTypeOfMember(matchArgsMemberInfo);
388
+ if (
389
+ isClassInstance(matchArgsType) &&
390
+ isTupleClass(matchArgsType) &&
391
+ !isUnboundedTupleClass(matchArgsType) &&
392
+ matchArgsType.tupleTypeArguments
393
+ ) {
394
+ const tupleArgs = matchArgsType.tupleTypeArguments;
395
+
396
+ // Are all the args string literals?
397
+ if (
398
+ tupleArgs.every(
399
+ (arg) =>
400
+ isClassInstance(arg.type) && ClassType.isBuiltIn(arg.type, 'str') && isLiteralType(arg.type)
401
+ )
402
+ ) {
403
+ return tupleArgs.map((arg) => (arg.type as ClassType).literalValue as string);
404
+ }
405
+ }
406
+ }
407
+
408
+ return [];
409
+ }
410
+
411
+ function narrowTypeBasedOnLiteralPattern(
412
+ evaluator: TypeEvaluator,
413
+ type: Type,
414
+ pattern: PatternLiteralNode,
415
+ isPositiveTest: boolean
416
+ ): Type {
417
+ const literalType = evaluator.getTypeOfExpression(pattern.expression).type;
418
+
419
+ if (!isPositiveTest) {
420
+ return mapSubtypes(type, (subtype) => {
421
+ if (
422
+ isClassInstance(literalType) &&
423
+ isLiteralType(literalType) &&
424
+ isClassInstance(subtype) &&
425
+ isLiteralType(subtype) &&
426
+ evaluator.assignType(literalType, subtype)
427
+ ) {
428
+ return undefined;
429
+ }
430
+
431
+ if (isNoneInstance(subtype) && isNoneInstance(literalType)) {
432
+ return undefined;
433
+ }
434
+
435
+ // Narrow a non-literal bool based on a literal bool pattern.
436
+ if (
437
+ isClassInstance(subtype) &&
438
+ ClassType.isBuiltIn(subtype, 'bool') &&
439
+ subtype.literalValue === undefined &&
440
+ isClassInstance(literalType) &&
441
+ ClassType.isBuiltIn(literalType, 'bool') &&
442
+ literalType.literalValue !== undefined
443
+ ) {
444
+ return ClassType.cloneWithLiteral(literalType, !(literalType.literalValue as boolean));
445
+ }
446
+
447
+ return subtype;
448
+ });
449
+ }
450
+
451
+ return mapSubtypes(type, (subtype) => {
452
+ if (evaluator.assignType(subtype, literalType)) {
453
+ return literalType;
454
+ }
455
+ return undefined;
456
+ });
457
+ }
458
+
459
+ function narrowTypeBasedOnClassPattern(
460
+ evaluator: TypeEvaluator,
461
+ type: Type,
462
+ pattern: PatternClassNode,
463
+ isPositiveTest: boolean
464
+ ): Type {
465
+ let exprType = evaluator.getTypeOfExpression(pattern.className, EvaluatorFlags.DoNotSpecialize).type;
466
+
467
+ // If this is a class (but not a type alias that refers to a class),
468
+ // specialize it with Unknown type arguments.
469
+ if (isClass(exprType) && !exprType.typeAliasInfo) {
470
+ exprType = specializeClassType(exprType);
471
+ }
472
+
473
+ if (!isPositiveTest) {
474
+ // Don't attempt to narrow if the class type is a more complex type (e.g. a TypeVar or union).
475
+ if (!isInstantiableClass(exprType)) {
476
+ return type;
477
+ }
478
+
479
+ let classType = exprType;
480
+
481
+ if (classType.details.typeParameters.length > 0) {
482
+ classType = ClassType.cloneForSpecialization(
483
+ classType,
484
+ /* typeArguments */ undefined,
485
+ /* isTypeArgumentExplicit */ false
486
+ );
487
+ }
488
+
489
+ const classInstance = convertToInstance(classType);
490
+ return evaluator.mapSubtypesExpandTypeVars(
491
+ type,
492
+ /* conditionFilter */ undefined,
493
+ (subjectSubtypeExpanded, subjectSubtypeUnexpanded) => {
494
+ if (!isClassInstance(subjectSubtypeExpanded)) {
495
+ return subjectSubtypeUnexpanded;
496
+ }
497
+
498
+ if (!evaluator.assignType(classInstance, subjectSubtypeExpanded)) {
499
+ return subjectSubtypeExpanded;
500
+ }
501
+
502
+ // If there are no arguments, we're done. We know that this match
503
+ // will never succeed.
504
+ if (pattern.arguments.length === 0) {
505
+ return undefined;
506
+ }
507
+
508
+ // We might be able to narrow further based on arguments, but only
509
+ // if the types match exactly or the subtype is a final class and
510
+ // therefore cannot be subclassed.
511
+ if (!evaluator.assignType(subjectSubtypeExpanded, classInstance)) {
512
+ if (!ClassType.isFinal(subjectSubtypeExpanded)) {
513
+ return subjectSubtypeExpanded;
514
+ }
515
+ }
516
+
517
+ // Are there any positional arguments? If so, try to get the mappings for
518
+ // these arguments by fetching the __match_args__ symbol from the class.
519
+ let positionalArgNames: string[] = [];
520
+ if (pattern.arguments.some((arg) => !arg.name)) {
521
+ if (isClass(subjectSubtypeExpanded)) {
522
+ positionalArgNames = getPositionalMatchArgNames(evaluator, subjectSubtypeExpanded);
523
+ }
524
+ }
525
+
526
+ for (let index = 0; index < pattern.arguments.length; index++) {
527
+ const narrowedArgType = narrowTypeOfClassPatternArgument(
528
+ evaluator,
529
+ pattern.arguments[index],
530
+ index,
531
+ positionalArgNames,
532
+ subjectSubtypeExpanded,
533
+ isPositiveTest
534
+ );
535
+
536
+ if (!isNever(narrowedArgType)) {
537
+ return subjectSubtypeUnexpanded;
538
+ }
539
+ }
540
+
541
+ // We've completely eliminated the type based on the arguments.
542
+ return undefined;
543
+ }
544
+ );
545
+ }
546
+
547
+ if (!TypeBase.isInstantiable(exprType) && !isNever(exprType)) {
548
+ evaluator.addDiagnostic(
549
+ getFileInfo(pattern).diagnosticRuleSet.reportGeneralTypeIssues,
550
+ DiagnosticRule.reportGeneralTypeIssues,
551
+ Localizer.DiagnosticAddendum.typeNotClass().format({ type: evaluator.printType(exprType) }),
552
+ pattern.className
553
+ );
554
+ return NeverType.createNever();
555
+ }
556
+
557
+ return evaluator.mapSubtypesExpandTypeVars(
558
+ exprType,
559
+ /* conditionFilter */ undefined,
560
+ (expandedSubtype, unexpandedSubtype) => {
561
+ if (isAnyOrUnknown(expandedSubtype)) {
562
+ return unexpandedSubtype;
563
+ }
564
+
565
+ if (isInstantiableClass(expandedSubtype)) {
566
+ return evaluator.mapSubtypesExpandTypeVars(
567
+ type,
568
+ /* conditionFilter */ undefined,
569
+ (subjectSubtypeExpanded) => {
570
+ if (isAnyOrUnknown(subjectSubtypeExpanded)) {
571
+ return convertToInstance(unexpandedSubtype);
572
+ }
573
+
574
+ if (isClassInstance(subjectSubtypeExpanded)) {
575
+ let resultType: Type;
576
+
577
+ if (
578
+ evaluator.assignType(
579
+ expandedSubtype,
580
+ ClassType.cloneAsInstantiable(subjectSubtypeExpanded)
581
+ )
582
+ ) {
583
+ resultType = subjectSubtypeExpanded;
584
+ } else if (
585
+ evaluator.assignType(
586
+ ClassType.cloneAsInstantiable(subjectSubtypeExpanded),
587
+ expandedSubtype
588
+ )
589
+ ) {
590
+ resultType = addConditionToType(
591
+ convertToInstance(unexpandedSubtype),
592
+ getTypeCondition(subjectSubtypeExpanded)
593
+ );
594
+
595
+ // Try to retain the type arguments for the pattern class type.
596
+ if (isInstantiableClass(unexpandedSubtype) && isClassInstance(subjectSubtypeExpanded)) {
597
+ if (
598
+ ClassType.isSpecialBuiltIn(unexpandedSubtype) ||
599
+ unexpandedSubtype.details.typeParameters.length > 0
600
+ ) {
601
+ const typeVarContext = new TypeVarContext(getTypeVarScopeId(unexpandedSubtype));
602
+ const unspecializedMatchType = ClassType.cloneForSpecialization(
603
+ unexpandedSubtype,
604
+ /* typeArguments */ undefined,
605
+ /* isTypeArgumentExplicit */ false
606
+ );
607
+
608
+ const matchTypeInstance = ClassType.cloneAsInstance(unspecializedMatchType);
609
+ if (
610
+ populateTypeVarContextBasedOnExpectedType(
611
+ evaluator,
612
+ matchTypeInstance,
613
+ subjectSubtypeExpanded,
614
+ typeVarContext,
615
+ []
616
+ )
617
+ ) {
618
+ resultType = applySolvedTypeVars(
619
+ matchTypeInstance,
620
+ typeVarContext,
621
+ /* unknownIfNotFound */ true
622
+ ) as ClassType;
623
+ }
624
+ }
625
+ }
626
+ } else {
627
+ return undefined;
628
+ }
629
+
630
+ // Are there any positional arguments? If so, try to get the mappings for
631
+ // these arguments by fetching the __match_args__ symbol from the class.
632
+ let positionalArgNames: string[] = [];
633
+ if (pattern.arguments.some((arg) => !arg.name)) {
634
+ positionalArgNames = getPositionalMatchArgNames(evaluator, expandedSubtype);
635
+ }
636
+
637
+ let isMatchValid = true;
638
+ pattern.arguments.forEach((arg, index) => {
639
+ // Narrow the arg pattern. It's possible that the actual type of the object
640
+ // being matched is a subtype of the resultType, so it might contain additional
641
+ // attributes that we don't know about.
642
+ const narrowedArgType = narrowTypeOfClassPatternArgument(
643
+ evaluator,
644
+ arg,
645
+ index,
646
+ positionalArgNames,
647
+ resultType,
648
+ isPositiveTest
649
+ );
650
+
651
+ if (isNever(narrowedArgType)) {
652
+ isMatchValid = false;
653
+ }
654
+ });
655
+
656
+ if (isMatchValid) {
657
+ return resultType;
658
+ }
659
+ }
660
+
661
+ return undefined;
662
+ }
663
+ );
664
+ }
665
+
666
+ return undefined;
667
+ }
668
+ );
669
+ }
670
+
671
+ // Narrows the pattern provided for a class pattern argument.
672
+ function narrowTypeOfClassPatternArgument(
673
+ evaluator: TypeEvaluator,
674
+ arg: PatternClassArgumentNode,
675
+ argIndex: number,
676
+ positionalArgNames: string[],
677
+ matchType: Type,
678
+ isPositiveTest: boolean
679
+ ) {
680
+ let argName: string | undefined;
681
+
682
+ if (arg.name) {
683
+ argName = arg.name.value;
684
+ } else if (argIndex < positionalArgNames.length) {
685
+ argName = positionalArgNames[argIndex];
686
+ }
687
+
688
+ if (isAnyOrUnknown(matchType)) {
689
+ return matchType;
690
+ }
691
+
692
+ if (!isClass(matchType)) {
693
+ return UnknownType.create();
694
+ }
695
+
696
+ const useSelfForPattern =
697
+ isClass(matchType) &&
698
+ classPatternSpecialCases.some((className) => matchType.details.fullName === className) &&
699
+ argIndex === 0 &&
700
+ !arg.name;
701
+
702
+ let argType: Type | undefined;
703
+ if (useSelfForPattern) {
704
+ argType = ClassType.cloneAsInstance(matchType);
705
+ } else {
706
+ if (argName) {
707
+ argType = evaluator.useSpeculativeMode(arg, () =>
708
+ // We need to apply a rather ugly cast here because PatternClassArgumentNode is
709
+ // not technically an ExpressionNode, but it is OK to use it in this context.
710
+ evaluator.getTypeOfObjectMember(
711
+ arg as any as ExpressionNode,
712
+ ClassType.cloneAsInstance(matchType),
713
+ argName!
714
+ )
715
+ )?.type;
716
+ }
717
+
718
+ if (!argType) {
719
+ if (!isPositiveTest) {
720
+ return matchType;
721
+ }
722
+
723
+ // If the class type in question is "final", we know that no additional
724
+ // attributes can be added by subtypes, so it's safe to eliminate this
725
+ // type entirely.
726
+ if (ClassType.isFinal(matchType)) {
727
+ return NeverType.createNever();
728
+ }
729
+
730
+ argType = UnknownType.create();
731
+ }
732
+ }
733
+
734
+ return narrowTypeBasedOnPattern(evaluator, argType, arg.pattern, isPositiveTest);
735
+ }
736
+
737
+ function narrowTypeBasedOnValuePattern(
738
+ evaluator: TypeEvaluator,
739
+ subjectType: Type,
740
+ pattern: PatternValueNode,
741
+ isPositiveTest: boolean
742
+ ): Type {
743
+ const valueType = evaluator.getTypeOfExpression(pattern.expression).type;
744
+ const narrowedSubtypes: Type[] = [];
745
+
746
+ evaluator.mapSubtypesExpandTypeVars(
747
+ valueType,
748
+ /* conditionFilter */ undefined,
749
+ (valueSubtypeExpanded, valueSubtypeUnexpanded) => {
750
+ narrowedSubtypes.push(
751
+ evaluator.mapSubtypesExpandTypeVars(
752
+ subjectType,
753
+ getTypeCondition(valueSubtypeExpanded),
754
+ (subjectSubtypeExpanded) => {
755
+ // If this is a negative test, see if it's an enum value.
756
+ if (!isPositiveTest) {
757
+ if (
758
+ isClassInstance(subjectSubtypeExpanded) &&
759
+ ClassType.isEnumClass(subjectSubtypeExpanded) &&
760
+ !isLiteralType(subjectSubtypeExpanded) &&
761
+ isClassInstance(valueSubtypeExpanded) &&
762
+ isSameWithoutLiteralValue(subjectSubtypeExpanded, valueSubtypeExpanded) &&
763
+ isLiteralType(valueSubtypeExpanded)
764
+ ) {
765
+ const allEnumTypes = enumerateLiteralsForType(evaluator, subjectSubtypeExpanded);
766
+ if (allEnumTypes) {
767
+ return combineTypes(
768
+ allEnumTypes.filter(
769
+ (enumType) => !ClassType.isLiteralValueSame(valueSubtypeExpanded, enumType)
770
+ )
771
+ );
772
+ }
773
+ } else if (
774
+ isClassInstance(subjectSubtypeExpanded) &&
775
+ isClassInstance(valueSubtypeExpanded) &&
776
+ ClassType.isLiteralValueSame(valueSubtypeExpanded, subjectSubtypeExpanded)
777
+ ) {
778
+ return undefined;
779
+ }
780
+
781
+ return subjectSubtypeExpanded;
782
+ }
783
+
784
+ if (isNever(valueSubtypeExpanded) || isNever(subjectSubtypeExpanded)) {
785
+ return NeverType.createNever();
786
+ }
787
+
788
+ if (isAnyOrUnknown(valueSubtypeExpanded) || isAnyOrUnknown(subjectSubtypeExpanded)) {
789
+ // If either type is "Unknown" (versus Any), propagate the Unknown.
790
+ return isUnknown(valueSubtypeExpanded) || isUnknown(subjectSubtypeExpanded)
791
+ ? UnknownType.create()
792
+ : AnyType.create();
793
+ }
794
+
795
+ // Determine if we assignment is supported for this combination of
796
+ // value subtype and matching subtype.
797
+ const returnType = evaluator.useSpeculativeMode(pattern.expression, () =>
798
+ evaluator.getTypeOfMagicMethodReturn(
799
+ valueSubtypeExpanded,
800
+ [subjectSubtypeExpanded],
801
+ '__eq__',
802
+ pattern.expression,
803
+ /* expectedType */ undefined
804
+ )
805
+ );
806
+
807
+ return returnType ? valueSubtypeUnexpanded : undefined;
808
+ }
809
+ )
810
+ );
811
+
812
+ return undefined;
813
+ }
814
+ );
815
+
816
+ return combineTypes(narrowedSubtypes);
817
+ }
818
+
819
+ // Returns information about all subtypes that match the definition of a "mapping" as
820
+ // specified in PEP 634.
821
+ function getMappingPatternInfo(evaluator: TypeEvaluator, type: Type): MappingPatternInfo[] {
822
+ const mappingInfo: MappingPatternInfo[] = [];
823
+
824
+ doForEachSubtype(type, (subtype) => {
825
+ const concreteSubtype = evaluator.makeTopLevelTypeVarsConcrete(subtype);
826
+
827
+ if (isAnyOrUnknown(concreteSubtype)) {
828
+ mappingInfo.push({
829
+ subtype,
830
+ dictTypeArgs: {
831
+ key: concreteSubtype,
832
+ value: concreteSubtype,
833
+ },
834
+ });
835
+ } else if (isClassInstance(concreteSubtype)) {
836
+ if (ClassType.isTypedDictClass(concreteSubtype)) {
837
+ mappingInfo.push({
838
+ subtype,
839
+ typedDict: concreteSubtype,
840
+ });
841
+ } else {
842
+ let mroClassToSpecialize: ClassType | undefined;
843
+ for (const mroClass of concreteSubtype.details.mro) {
844
+ if (isInstantiableClass(mroClass) && ClassType.isBuiltIn(mroClass, 'Mapping')) {
845
+ mroClassToSpecialize = mroClass;
846
+ break;
847
+ }
848
+ }
849
+
850
+ if (mroClassToSpecialize) {
851
+ const specializedMapping = partiallySpecializeType(
852
+ mroClassToSpecialize,
853
+ concreteSubtype
854
+ ) as ClassType;
855
+ if (specializedMapping.typeArguments && specializedMapping.typeArguments.length >= 2) {
856
+ mappingInfo.push({
857
+ subtype,
858
+ dictTypeArgs: {
859
+ key: specializedMapping.typeArguments[0],
860
+ value: specializedMapping.typeArguments[1],
861
+ },
862
+ });
863
+ }
864
+ }
865
+ }
866
+ }
867
+ });
868
+
869
+ return mappingInfo;
870
+ }
871
+
872
+ // Returns information about all subtypes that match the definition of a "sequence" as
873
+ // specified in PEP 634. For types that are not sequences or sequences that are not of
874
+ // sufficient length, it sets definiteNoMatch to true.
875
+ function getSequencePatternInfo(
876
+ evaluator: TypeEvaluator,
877
+ type: Type,
878
+ entryCount: number,
879
+ starEntryIndex: number | undefined
880
+ ): SequencePatternInfo[] {
881
+ const sequenceInfo: SequencePatternInfo[] = [];
882
+ const minEntryCount = starEntryIndex === undefined ? entryCount : entryCount - 1;
883
+
884
+ doForEachSubtype(type, (subtype) => {
885
+ const concreteSubtype = evaluator.makeTopLevelTypeVarsConcrete(subtype);
886
+ let mroClassToSpecialize: ClassType | undefined;
887
+ let pushedEntry = false;
888
+
889
+ if (isAnyOrUnknown(concreteSubtype)) {
890
+ sequenceInfo.push({
891
+ subtype,
892
+ entryTypes: [concreteSubtype],
893
+ isIndeterminateLength: true,
894
+ definiteNoMatch: false,
895
+ });
896
+ return;
897
+ }
898
+
899
+ if (isClassInstance(concreteSubtype)) {
900
+ if (ClassType.isBuiltIn(concreteSubtype, 'object')) {
901
+ sequenceInfo.push({
902
+ subtype,
903
+ entryTypes: [convertToInstance(concreteSubtype)],
904
+ isIndeterminateLength: true,
905
+ isObject: true,
906
+ definiteNoMatch: false,
907
+ });
908
+ return;
909
+ }
910
+
911
+ for (const mroClass of concreteSubtype.details.mro) {
912
+ if (!isInstantiableClass(mroClass)) {
913
+ break;
914
+ }
915
+
916
+ // Strings, bytes, and bytearray are explicitly excluded.
917
+ if (
918
+ ClassType.isBuiltIn(mroClass, 'str') ||
919
+ ClassType.isBuiltIn(mroClass, 'bytes') ||
920
+ ClassType.isBuiltIn(mroClass, 'bytearray')
921
+ ) {
922
+ break;
923
+ }
924
+
925
+ if (ClassType.isBuiltIn(mroClass, 'Sequence')) {
926
+ mroClassToSpecialize = mroClass;
927
+ break;
928
+ }
929
+
930
+ if (isTupleClass(mroClass)) {
931
+ mroClassToSpecialize = mroClass;
932
+ break;
933
+ }
934
+ }
935
+
936
+ if (mroClassToSpecialize) {
937
+ const specializedSequence = partiallySpecializeType(mroClassToSpecialize, concreteSubtype) as ClassType;
938
+
939
+ if (isTupleClass(specializedSequence)) {
940
+ if (specializedSequence.tupleTypeArguments) {
941
+ if (isUnboundedTupleClass(specializedSequence)) {
942
+ sequenceInfo.push({
943
+ subtype,
944
+ entryTypes: [combineTypes(specializedSequence.tupleTypeArguments.map((t) => t.type))],
945
+ isIndeterminateLength: true,
946
+ isTuple: true,
947
+ definiteNoMatch: false,
948
+ });
949
+ pushedEntry = true;
950
+ } else {
951
+ if (
952
+ specializedSequence.tupleTypeArguments.length >= minEntryCount &&
953
+ (starEntryIndex !== undefined ||
954
+ specializedSequence.tupleTypeArguments.length === minEntryCount)
955
+ ) {
956
+ sequenceInfo.push({
957
+ subtype,
958
+ entryTypes: specializedSequence.tupleTypeArguments.map((t) => t.type),
959
+ isIndeterminateLength: false,
960
+ isTuple: true,
961
+ definiteNoMatch: false,
962
+ });
963
+ pushedEntry = true;
964
+ }
965
+ }
966
+ }
967
+ } else {
968
+ sequenceInfo.push({
969
+ subtype,
970
+ entryTypes: [
971
+ specializedSequence.typeArguments && specializedSequence.typeArguments.length > 0
972
+ ? specializedSequence.typeArguments[0]
973
+ : UnknownType.create(),
974
+ ],
975
+ isIndeterminateLength: true,
976
+ definiteNoMatch: false,
977
+ });
978
+ pushedEntry = true;
979
+ }
980
+ }
981
+ }
982
+
983
+ // Push an entry that indicates that this is definitely not a match.
984
+ if (!pushedEntry) {
985
+ sequenceInfo.push({
986
+ subtype,
987
+ entryTypes: [],
988
+ isIndeterminateLength: true,
989
+ definiteNoMatch: true,
990
+ });
991
+ }
992
+ });
993
+
994
+ return sequenceInfo;
995
+ }
996
+
997
+ function getTypeOfPatternSequenceEntry(
998
+ evaluator: TypeEvaluator,
999
+ node: ParseNode,
1000
+ sequenceInfo: SequencePatternInfo,
1001
+ entryIndex: number,
1002
+ entryCount: number,
1003
+ starEntryIndex: number | undefined,
1004
+ unpackStarEntry: boolean,
1005
+ isSubjectObject: boolean
1006
+ ): Type {
1007
+ if (sequenceInfo.isIndeterminateLength) {
1008
+ let entryType = sequenceInfo.entryTypes[0];
1009
+
1010
+ // If the subject is typed as an "object", then the star entry
1011
+ // is simply a list[object]. Without this special case, the list
1012
+ // will be typed based on the union of all elements in the sequence.
1013
+ if (isSubjectObject) {
1014
+ const objectType = evaluator.getBuiltInObject(node, 'object');
1015
+ if (objectType && isClassInstance(objectType)) {
1016
+ entryType = objectType;
1017
+ }
1018
+ }
1019
+
1020
+ if (!unpackStarEntry && entryIndex === starEntryIndex && !isNever(entryType)) {
1021
+ entryType = wrapTypeInList(evaluator, node, entryType);
1022
+ }
1023
+
1024
+ return entryType;
1025
+ }
1026
+
1027
+ if (starEntryIndex === undefined || entryIndex < starEntryIndex) {
1028
+ return sequenceInfo.entryTypes[entryIndex];
1029
+ }
1030
+
1031
+ if (entryIndex === starEntryIndex) {
1032
+ // Create a list out of the entries that map to the star entry.
1033
+ // Note that we strip literal types here.
1034
+ const starEntryTypes = sequenceInfo.entryTypes
1035
+ .slice(starEntryIndex, starEntryIndex + sequenceInfo.entryTypes.length - entryCount + 1)
1036
+ .map((type) => stripLiteralValue(type));
1037
+
1038
+ let entryType = combineTypes(starEntryTypes);
1039
+
1040
+ if (!unpackStarEntry) {
1041
+ entryType = wrapTypeInList(evaluator, node, entryType);
1042
+ }
1043
+
1044
+ return entryType;
1045
+ }
1046
+
1047
+ // The entry index is past the index of the star entry, so we need
1048
+ // to index from the end of the sequence rather than the start.
1049
+ const itemIndex = sequenceInfo.entryTypes.length - (entryCount - entryIndex);
1050
+ assert(itemIndex >= 0 && itemIndex < sequenceInfo.entryTypes.length);
1051
+
1052
+ return sequenceInfo.entryTypes[itemIndex];
1053
+ }
1054
+
1055
+ // Recursively assigns the specified type to the pattern and any capture
1056
+ // nodes within it
1057
+ export function assignTypeToPatternTargets(
1058
+ evaluator: TypeEvaluator,
1059
+ type: Type,
1060
+ isTypeIncomplete: boolean,
1061
+ isSubjectObject: boolean,
1062
+ pattern: PatternAtomNode
1063
+ ) {
1064
+ // Further narrow the type based on this pattern.
1065
+ type = narrowTypeBasedOnPattern(evaluator, type, pattern, /* positiveTest */ true);
1066
+
1067
+ switch (pattern.nodeType) {
1068
+ case ParseNodeType.PatternSequence: {
1069
+ const sequenceInfo = getSequencePatternInfo(
1070
+ evaluator,
1071
+ type,
1072
+ pattern.entries.length,
1073
+ pattern.starEntryIndex
1074
+ ).filter((seqInfo) => !seqInfo.definiteNoMatch);
1075
+
1076
+ pattern.entries.forEach((entry, index) => {
1077
+ const entryType = combineTypes(
1078
+ sequenceInfo.map((info) =>
1079
+ getTypeOfPatternSequenceEntry(
1080
+ evaluator,
1081
+ pattern,
1082
+ info,
1083
+ index,
1084
+ pattern.entries.length,
1085
+ pattern.starEntryIndex,
1086
+ /* unpackStarEntry */ false,
1087
+ isSubjectObject
1088
+ )
1089
+ )
1090
+ );
1091
+
1092
+ assignTypeToPatternTargets(evaluator, entryType, isTypeIncomplete, /* isSubjectObject */ false, entry);
1093
+ });
1094
+ break;
1095
+ }
1096
+
1097
+ case ParseNodeType.PatternAs: {
1098
+ if (pattern.target) {
1099
+ evaluator.assignTypeToExpression(pattern.target, type, isTypeIncomplete, pattern.target);
1100
+ }
1101
+
1102
+ pattern.orPatterns.forEach((orPattern) => {
1103
+ assignTypeToPatternTargets(evaluator, type, isTypeIncomplete, isSubjectObject, orPattern);
1104
+
1105
+ // OR patterns are evaluated left to right, so we can narrow
1106
+ // the type as we go.
1107
+ type = narrowTypeBasedOnPattern(evaluator, type, orPattern, /* positiveTest */ false);
1108
+ });
1109
+ break;
1110
+ }
1111
+
1112
+ case ParseNodeType.PatternCapture: {
1113
+ evaluator.assignTypeToExpression(
1114
+ pattern.target,
1115
+ pattern.isWildcard ? AnyType.create() : type,
1116
+ isTypeIncomplete,
1117
+ pattern.target
1118
+ );
1119
+ break;
1120
+ }
1121
+
1122
+ case ParseNodeType.PatternMapping: {
1123
+ const mappingInfo = getMappingPatternInfo(evaluator, type);
1124
+
1125
+ pattern.entries.forEach((mappingEntry) => {
1126
+ const keyTypes: Type[] = [];
1127
+ const valueTypes: Type[] = [];
1128
+
1129
+ mappingInfo.forEach((mappingSubtypeInfo) => {
1130
+ if (mappingSubtypeInfo.typedDict) {
1131
+ if (mappingEntry.nodeType === ParseNodeType.PatternMappingKeyEntry) {
1132
+ const keyType = narrowTypeBasedOnPattern(
1133
+ evaluator,
1134
+ evaluator.getBuiltInObject(pattern, 'str'),
1135
+ mappingEntry.keyPattern,
1136
+ /* isPositiveTest */ true
1137
+ );
1138
+ keyTypes.push(keyType);
1139
+
1140
+ doForEachSubtype(keyType, (keySubtype) => {
1141
+ if (
1142
+ isClassInstance(keySubtype) &&
1143
+ ClassType.isBuiltIn(keySubtype, 'str') &&
1144
+ isLiteralType(keySubtype)
1145
+ ) {
1146
+ const tdEntries = getTypedDictMembersForClass(
1147
+ evaluator,
1148
+ mappingSubtypeInfo.typedDict!
1149
+ );
1150
+ const valueInfo = tdEntries.get(keySubtype.literalValue as string);
1151
+ valueTypes.push(valueInfo ? valueInfo.valueType : UnknownType.create());
1152
+ } else {
1153
+ valueTypes.push(UnknownType.create());
1154
+ }
1155
+ });
1156
+ } else if (mappingEntry.nodeType === ParseNodeType.PatternMappingExpandEntry) {
1157
+ keyTypes.push(evaluator.getBuiltInObject(pattern, 'str'));
1158
+ valueTypes.push(UnknownType.create());
1159
+ }
1160
+ } else if (mappingSubtypeInfo.dictTypeArgs) {
1161
+ if (mappingEntry.nodeType === ParseNodeType.PatternMappingKeyEntry) {
1162
+ const keyType = narrowTypeBasedOnPattern(
1163
+ evaluator,
1164
+ mappingSubtypeInfo.dictTypeArgs.key,
1165
+ mappingEntry.keyPattern,
1166
+ /* isPositiveTest */ true
1167
+ );
1168
+ keyTypes.push(keyType);
1169
+ valueTypes.push(
1170
+ narrowTypeBasedOnPattern(
1171
+ evaluator,
1172
+ mappingSubtypeInfo.dictTypeArgs.value,
1173
+ mappingEntry.valuePattern,
1174
+ /* isPositiveTest */ true
1175
+ )
1176
+ );
1177
+ } else if (mappingEntry.nodeType === ParseNodeType.PatternMappingExpandEntry) {
1178
+ keyTypes.push(mappingSubtypeInfo.dictTypeArgs.key);
1179
+ valueTypes.push(mappingSubtypeInfo.dictTypeArgs.value);
1180
+ }
1181
+ }
1182
+ });
1183
+
1184
+ const keyType = combineTypes(keyTypes);
1185
+ const valueType = combineTypes(valueTypes);
1186
+
1187
+ if (mappingEntry.nodeType === ParseNodeType.PatternMappingKeyEntry) {
1188
+ assignTypeToPatternTargets(
1189
+ evaluator,
1190
+ keyType,
1191
+ isTypeIncomplete,
1192
+ /* isSubjectObject */ false,
1193
+ mappingEntry.keyPattern
1194
+ );
1195
+ assignTypeToPatternTargets(
1196
+ evaluator,
1197
+ valueType,
1198
+ isTypeIncomplete,
1199
+ /* isSubjectObject */ false,
1200
+ mappingEntry.valuePattern
1201
+ );
1202
+ } else if (mappingEntry.nodeType === ParseNodeType.PatternMappingExpandEntry) {
1203
+ const dictClass = evaluator.getBuiltInType(pattern, 'dict');
1204
+ const strType = evaluator.getBuiltInObject(pattern, 'str');
1205
+ const dictType =
1206
+ dictClass && isInstantiableClass(dictClass) && isClassInstance(strType)
1207
+ ? ClassType.cloneAsInstance(
1208
+ ClassType.cloneForSpecialization(
1209
+ dictClass,
1210
+ [keyType, valueType],
1211
+ /* isTypeArgumentExplicit */ true
1212
+ )
1213
+ )
1214
+ : UnknownType.create();
1215
+ evaluator.assignTypeToExpression(
1216
+ mappingEntry.target,
1217
+ dictType,
1218
+ isTypeIncomplete,
1219
+ mappingEntry.target
1220
+ );
1221
+ }
1222
+ });
1223
+ break;
1224
+ }
1225
+
1226
+ case ParseNodeType.PatternClass: {
1227
+ const argTypes: Type[][] = pattern.arguments.map((arg) => []);
1228
+
1229
+ evaluator.mapSubtypesExpandTypeVars(type, /* conditionFilter */ undefined, (expandedSubtype) => {
1230
+ if (isClassInstance(expandedSubtype)) {
1231
+ doForEachSubtype(type, (subjectSubtype) => {
1232
+ const concreteSubtype = evaluator.makeTopLevelTypeVarsConcrete(subjectSubtype);
1233
+
1234
+ if (isAnyOrUnknown(concreteSubtype)) {
1235
+ pattern.arguments.forEach((arg, index) => {
1236
+ argTypes[index].push(concreteSubtype);
1237
+ });
1238
+ } else if (isClassInstance(concreteSubtype)) {
1239
+ // Are there any positional arguments? If so, try to get the mappings for
1240
+ // these arguments by fetching the __match_args__ symbol from the class.
1241
+ let positionalArgNames: string[] = [];
1242
+ if (pattern.arguments.some((arg) => !arg.name)) {
1243
+ positionalArgNames = getPositionalMatchArgNames(
1244
+ evaluator,
1245
+ ClassType.cloneAsInstantiable(expandedSubtype)
1246
+ );
1247
+ }
1248
+
1249
+ pattern.arguments.forEach((arg, index) => {
1250
+ const narrowedArgType = narrowTypeOfClassPatternArgument(
1251
+ evaluator,
1252
+ arg,
1253
+ index,
1254
+ positionalArgNames,
1255
+ ClassType.cloneAsInstantiable(expandedSubtype),
1256
+ /* isPositiveTest */ true
1257
+ );
1258
+ argTypes[index].push(narrowedArgType);
1259
+ });
1260
+ }
1261
+ });
1262
+ } else {
1263
+ pattern.arguments.forEach((arg, index) => {
1264
+ argTypes[index].push(UnknownType.create());
1265
+ });
1266
+ }
1267
+
1268
+ return undefined;
1269
+ });
1270
+
1271
+ pattern.arguments.forEach((arg, index) => {
1272
+ assignTypeToPatternTargets(
1273
+ evaluator,
1274
+ combineTypes(argTypes[index]),
1275
+ isTypeIncomplete,
1276
+ /* isSubjectObject */ false,
1277
+ arg.pattern
1278
+ );
1279
+ });
1280
+ break;
1281
+ }
1282
+
1283
+ case ParseNodeType.PatternLiteral:
1284
+ case ParseNodeType.PatternValue:
1285
+ case ParseNodeType.Error: {
1286
+ // Nothing to do here.
1287
+ break;
1288
+ }
1289
+ }
1290
+ }
1291
+
1292
+ function wrapTypeInList(evaluator: TypeEvaluator, node: ParseNode, type: Type): Type {
1293
+ if (isNever(type)) {
1294
+ return type;
1295
+ }
1296
+
1297
+ const listObjectType = convertToInstance(evaluator.getBuiltInObject(node, 'list'));
1298
+ if (listObjectType && isClassInstance(listObjectType)) {
1299
+ return ClassType.cloneForSpecialization(listObjectType, [type], /* isTypeArgumentExplicit */ true);
1300
+ }
1301
+
1302
+ return UnknownType.create();
1303
+ }
1304
+
1305
+ export function validateClassPattern(evaluator: TypeEvaluator, pattern: PatternClassNode) {
1306
+ const exprType = evaluator.getTypeOfExpression(pattern.className, EvaluatorFlags.DoNotSpecialize).type;
1307
+
1308
+ if (isAnyOrUnknown(exprType)) {
1309
+ return;
1310
+ }
1311
+
1312
+ // Check for certain uses of type aliases that generate runtime exceptions.
1313
+ if (
1314
+ exprType.typeAliasInfo &&
1315
+ isInstantiableClass(exprType) &&
1316
+ exprType.typeArguments &&
1317
+ exprType.isTypeArgumentExplicit
1318
+ ) {
1319
+ evaluator.addDiagnostic(
1320
+ getFileInfo(pattern).diagnosticRuleSet.reportGeneralTypeIssues,
1321
+ DiagnosticRule.reportGeneralTypeIssues,
1322
+ Localizer.Diagnostic.classPatternTypeAlias().format({ type: evaluator.printType(exprType) }),
1323
+ pattern.className
1324
+ );
1325
+ } else if (!isInstantiableClass(exprType)) {
1326
+ if (!isNever(exprType)) {
1327
+ evaluator.addDiagnostic(
1328
+ getFileInfo(pattern).diagnosticRuleSet.reportGeneralTypeIssues,
1329
+ DiagnosticRule.reportGeneralTypeIssues,
1330
+ Localizer.DiagnosticAddendum.typeNotClass().format({ type: evaluator.printType(exprType) }),
1331
+ pattern.className
1332
+ );
1333
+ }
1334
+ } else {
1335
+ const isBuiltIn = classPatternSpecialCases.some((className) => exprType.details.fullName === className);
1336
+
1337
+ // If it's a special-case builtin class, only one positional argument is allowed.
1338
+ if (isBuiltIn) {
1339
+ if (pattern.arguments.length > 1) {
1340
+ evaluator.addDiagnostic(
1341
+ getFileInfo(pattern).diagnosticRuleSet.reportGeneralTypeIssues,
1342
+ DiagnosticRule.reportGeneralTypeIssues,
1343
+ Localizer.Diagnostic.classPatternBuiltInArgCount(),
1344
+ pattern.arguments[1]
1345
+ );
1346
+ } else if (pattern.arguments.length === 1 && pattern.arguments[0].name) {
1347
+ evaluator.addDiagnostic(
1348
+ getFileInfo(pattern).diagnosticRuleSet.reportGeneralTypeIssues,
1349
+ DiagnosticRule.reportGeneralTypeIssues,
1350
+ Localizer.Diagnostic.classPatternBuiltInArgPositional(),
1351
+ pattern.arguments[0].name
1352
+ );
1353
+ }
1354
+ }
1355
+ }
1356
+ }