@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,1361 @@
1
+ /*
2
+ * importAdder.test.ts
3
+ * Copyright (c) Microsoft Corporation.
4
+ * Licensed under the MIT license.
5
+ *
6
+ * tests for importMover.
7
+ */
8
+
9
+ import assert from 'assert';
10
+ import { CancellationToken } from 'vscode-languageserver';
11
+
12
+ import { rangesAreEqual, TextRange } from '../common/textRange';
13
+ import { ImportAdder } from '../languageService/importAdder';
14
+ import { parseAndGetTestState } from './harness/fourslash/testState';
15
+
16
+ test('builtin types', () => {
17
+ const code = `
18
+ // @filename: test1.py
19
+ //// [|/*src*/a: str = "hello"
20
+ //// b: int = 1
21
+ //// c: True = True
22
+ //// d: None = None|]
23
+
24
+ // @filename: test2.py
25
+ //// [|/*dest*/|]
26
+ `;
27
+
28
+ testImportMove(code);
29
+ });
30
+
31
+ test('intrinsic types', () => {
32
+ const code = `
33
+ // @filename: test1.py
34
+ //// if __name__ == __path__:
35
+ //// pass
36
+ ////
37
+ //// [|/*src*/if __name__ === "__main__":
38
+ //// pass
39
+ //// b = __path__|]
40
+
41
+ // @filename: test2.py
42
+ //// [|/*dest*/|]
43
+ `;
44
+
45
+ testImportMove(code);
46
+ });
47
+
48
+ test('handle variable in range', () => {
49
+ const code = `
50
+ // @filename: test1.py
51
+ //// [|/*src*/variableToMove = 1|]
52
+
53
+ // @filename: test2.py
54
+ //// [|/*dest*/|]
55
+ `;
56
+
57
+ testImportMove(code);
58
+ });
59
+
60
+ test('move variable', () => {
61
+ const code = `
62
+ // @filename: test1.py
63
+ //// variableToMove = 1
64
+ //// [|/*src*/a = variableToMove|]
65
+
66
+ // @filename: test2.py
67
+ //// [|/*dest*/{|"r":"from test1 import variableToMove!n!!n!!n!"|}|]
68
+ `;
69
+
70
+ testImportMove(code);
71
+ });
72
+
73
+ test('move multiple variables', () => {
74
+ const code = `
75
+ // @filename: test1.py
76
+ //// variableToMove1 = 1
77
+ //// variableToMove2 = 2
78
+ //// [|/*src*/a = variableToMove1
79
+ //// a = variableToMove2|]
80
+
81
+ // @filename: test2.py
82
+ //// [|/*dest*/{|"r":"from test1 import variableToMove1, variableToMove2!n!!n!!n!"|}|]
83
+ `;
84
+
85
+ testImportMove(code);
86
+ });
87
+
88
+ test('handle local variables', () => {
89
+ const code = `
90
+ // @filename: test1.py
91
+ //// def foo():
92
+ //// variableToMove1 = 1
93
+ //// variableToMove2 = 2
94
+ //// [|/*src*/a = variableToMove1
95
+ //// a = variableToMove2|]
96
+
97
+ // @filename: test2.py
98
+ //// [|/*dest*/|]
99
+ `;
100
+
101
+ testImportMove(code);
102
+ });
103
+
104
+ test('handle parameter variable', () => {
105
+ const code = `
106
+ // @filename: test1.py
107
+ //// def foo(p: int):
108
+ //// [|/*src*/a = p|]
109
+
110
+ // @filename: test2.py
111
+ //// [|/*dest*/|]
112
+ `;
113
+
114
+ testImportMove(code);
115
+ });
116
+
117
+ test('move private variable', () => {
118
+ const code = `
119
+ // @filename: test1.py
120
+ //// __private = 1
121
+ //// [|/*src*/a = __private|]
122
+
123
+ // @filename: test2.py
124
+ //// [|/*dest*/{|"r":"from test1 import __private!n!!n!!n!"|}|]
125
+ `;
126
+
127
+ testImportMove(code);
128
+ });
129
+
130
+ test('handle function in range', () => {
131
+ const code = `
132
+ // @filename: test1.py
133
+ //// [|/*src*/def foo():
134
+ //// pass|]
135
+
136
+ // @filename: test2.py
137
+ //// [|/*dest*/|]
138
+ `;
139
+
140
+ testImportMove(code);
141
+ });
142
+
143
+ test('move function', () => {
144
+ const code = `
145
+ // @filename: test1.py
146
+ //// def foo(): pass
147
+ //// [|/*src*/foo()|]
148
+
149
+ // @filename: test2.py
150
+ //// [|/*dest*/{|"r":"from test1 import foo!n!!n!!n!"|}|]
151
+ `;
152
+
153
+ testImportMove(code);
154
+ });
155
+
156
+ test('move multiple functions', () => {
157
+ const code = `
158
+ // @filename: test1.py
159
+ //// def foo(): pass
160
+ //// def bar(): pass
161
+ //// [|/*src*/foo()
162
+ //// bar()|]
163
+
164
+ // @filename: test2.py
165
+ //// [|/*dest*/{|"r":"from test1 import bar, foo!n!!n!!n!"|}|]
166
+ `;
167
+
168
+ testImportMove(code);
169
+ });
170
+
171
+ test('handle inner function', () => {
172
+ const code = `
173
+ // @filename: test1.py
174
+ //// def foo():
175
+ //// def bar(): pass
176
+ //// [|/*src*/bar()|]
177
+
178
+ // @filename: test2.py
179
+ //// [|/*dest*/|]
180
+ `;
181
+
182
+ testImportMove(code);
183
+ });
184
+
185
+ test('move private function', () => {
186
+ const code = `
187
+ // @filename: test1.py
188
+ //// def __private(): pass
189
+ //// [|/*src*/__private()|]
190
+
191
+ // @filename: test2.py
192
+ //// [|/*dest*/{|"r":"from test1 import __private!n!!n!!n!"|}|]
193
+ `;
194
+
195
+ testImportMove(code);
196
+ });
197
+
198
+ test('handle class in range', () => {
199
+ const code = `
200
+ // @filename: test1.py
201
+ //// [|/*src*/class A: pass|]
202
+
203
+ // @filename: test2.py
204
+ //// [|/*dest*/|]
205
+ `;
206
+
207
+ testImportMove(code);
208
+ });
209
+
210
+ test('move class', () => {
211
+ const code = `
212
+ // @filename: test1.py
213
+ //// class A: pass
214
+ //// [|/*src*/a = A()|]
215
+
216
+ // @filename: test2.py
217
+ //// [|/*dest*/{|"r":"from test1 import A!n!!n!!n!"|}|]
218
+ `;
219
+
220
+ testImportMove(code);
221
+ });
222
+
223
+ test('move multiple classes', () => {
224
+ const code = `
225
+ // @filename: test1.py
226
+ //// class A: pass
227
+ //// class B: pass
228
+ //// [|/*src*/a = A()
229
+ //// a = B()|]
230
+
231
+ // @filename: test2.py
232
+ //// [|/*dest*/{|"r":"from test1 import A, B!n!!n!!n!"|}|]
233
+ `;
234
+
235
+ testImportMove(code);
236
+ });
237
+
238
+ test('handle inner class through self', () => {
239
+ const code = `
240
+ // @filename: test1.py
241
+ //// class A:
242
+ //// class B: pass
243
+ //// def foo(self):
244
+ //// [|/*src*/b = self.B()|]
245
+
246
+ // @filename: test2.py
247
+ //// [|/*dest*/|]
248
+ `;
249
+
250
+ testImportMove(code);
251
+ });
252
+
253
+ test('handle inner class through type', () => {
254
+ const code = `
255
+ // @filename: test1.py
256
+ //// class A:
257
+ //// class B: pass
258
+ //// def foo(self):
259
+ //// [|/*src*/b = A.B()|]
260
+
261
+ // @filename: test2.py
262
+ //// [|/*dest*/{|"r":"from test1 import A!n!!n!!n!"|}|]
263
+ `;
264
+
265
+ testImportMove(code);
266
+ });
267
+
268
+ test('handle class variable', () => {
269
+ const code = `
270
+ // @filename: test1.py
271
+ //// class A:
272
+ //// def __init__(self):
273
+ //// self.a = 1
274
+ //// c = A();
275
+ //// [|/*src*/a = c.a|]
276
+
277
+ // @filename: test2.py
278
+ //// [|/*dest*/{|"r":"from test1 import c!n!!n!!n!"|}|]
279
+ `;
280
+
281
+ testImportMove(code);
282
+ });
283
+
284
+ test('handle class static variable', () => {
285
+ const code = `
286
+ // @filename: test1.py
287
+ //// class A:
288
+ //// V = 1
289
+ //// [|/*src*/a = A.V|]
290
+
291
+ // @filename: test2.py
292
+ //// [|/*dest*/{|"r":"from test1 import A!n!!n!!n!"|}|]
293
+ `;
294
+
295
+ testImportMove(code);
296
+ });
297
+
298
+ test('handle class function', () => {
299
+ const code = `
300
+ // @filename: test1.py
301
+ //// class A:
302
+ //// def __init__(self): pass
303
+ //// def foo(self): pass
304
+ //// c = A();
305
+ //// [|/*src*/a = c.foo()|]
306
+
307
+ // @filename: test2.py
308
+ //// [|/*dest*/{|"r":"from test1 import c!n!!n!!n!"|}|]
309
+ `;
310
+
311
+ testImportMove(code);
312
+ });
313
+
314
+ test('handle class static function', () => {
315
+ const code = `
316
+ // @filename: test1.py
317
+ //// class A:
318
+ //// def Foo(): pass
319
+ //// [|/*src*/a = A.Foo()|]
320
+
321
+ // @filename: test2.py
322
+ //// [|/*dest*/{|"r":"from test1 import A!n!!n!!n!"|}|]
323
+ `;
324
+
325
+ testImportMove(code);
326
+ });
327
+
328
+ test('handle class function parameter', () => {
329
+ const code = `
330
+ // @filename: test1.py
331
+ //// class A:
332
+ //// def __init__(self):
333
+ //// [|/*src*/self.a = 1|]
334
+
335
+ // @filename: test2.py
336
+ //// [|/*dest*/|]
337
+ `;
338
+
339
+ testImportMove(code);
340
+ });
341
+
342
+ test('move private class', () => {
343
+ const code = `
344
+ // @filename: test1.py
345
+ //// class __A:
346
+ //// class B: pass
347
+ //// [|/*src*/a = __A()|]
348
+
349
+ // @filename: test2.py
350
+ //// [|/*dest*/{|"r":"from test1 import __A!n!!n!!n!"|}|]
351
+ `;
352
+
353
+ testImportMove(code);
354
+ });
355
+
356
+ test('move simple import statement', () => {
357
+ const code = `
358
+ // @filename: test1.py
359
+ //// import typing
360
+ ////
361
+ //// [|/*src*/a: typing.Any = 1|]
362
+
363
+ // @filename: test2.py
364
+ //// [|/*dest*/{|"r":"import typing!n!!n!!n!"|}|]
365
+ ////
366
+ `;
367
+
368
+ testImportMove(code);
369
+ });
370
+
371
+ test('move import statement with alias', () => {
372
+ const code = `
373
+ // @filename: test1.py
374
+ //// import typing as t
375
+ ////
376
+ //// [|/*src*/a: t.Any = 1|]
377
+
378
+ // @filename: test2.py
379
+ //// [|/*dest*/{|"r":"import typing as t!n!!n!!n!"|}|]
380
+ ////
381
+ `;
382
+
383
+ testImportMove(code);
384
+ });
385
+
386
+ test('move dotted import statement', () => {
387
+ const code = `
388
+ // @filename: test1.py
389
+ //// import json.encoder
390
+ ////
391
+ //// [|/*src*/a = json.encoder.JSONEncoder()|]
392
+
393
+ // @filename: test2.py
394
+ //// [|/*dest*/{|"r":"import json.encoder!n!!n!!n!"|}|]
395
+ ////
396
+ `;
397
+
398
+ testImportMove(code);
399
+ });
400
+
401
+ test('move dotted statement with alias', () => {
402
+ const code = `
403
+ // @filename: test1.py
404
+ //// import json.encoder as j
405
+ ////
406
+ //// [|/*src*/a = j.JSONEncoder()|]
407
+
408
+ // @filename: test2.py
409
+ //// [|/*dest*/{|"r":"import json.encoder as j!n!!n!!n!"|}|]
410
+ ////
411
+ `;
412
+
413
+ testImportMove(code);
414
+ });
415
+
416
+ test('move both dotted import and regular statement', () => {
417
+ const code = `
418
+ // @filename: test1.py
419
+ //// import json
420
+ //// import json.encoder
421
+ ////
422
+ //// a = json.encoder.JSONEncoder()
423
+ //// [|/*src*/b = json.loads("")|]
424
+
425
+ // @filename: test2.py
426
+ //// [|/*dest*/{|"r":"import json!n!!n!!n!"|}|]
427
+ ////
428
+ `;
429
+
430
+ testImportMove(code);
431
+ });
432
+
433
+ test('move both dotted import and regular statement with alias', () => {
434
+ const code = `
435
+ // @filename: test1.py
436
+ //// import json as j
437
+ //// import json.encoder
438
+ ////
439
+ //// a = json.encoder.JSONEncoder()
440
+ //// [|/*src*/b = j.loads("")|]
441
+
442
+ // @filename: test2.py
443
+ //// [|/*dest*/{|"r":"import json as j!n!!n!!n!"|}|]
444
+ ////
445
+ `;
446
+
447
+ testImportMove(code);
448
+ });
449
+
450
+ test('move multiple import statements', () => {
451
+ const code = `
452
+ // @filename: test1.py
453
+ //// import json
454
+ //// import json.encoder
455
+ ////
456
+ //// [|/*src*/a = json.encoder.JSONEncoder()
457
+ //// b = json.loads("")|]
458
+
459
+ // @filename: test2.py
460
+ //// [|/*dest*/{|"r":"import json!n!import json.encoder!n!!n!!n!"|}|]
461
+ ////
462
+ `;
463
+
464
+ testImportMove(code);
465
+ });
466
+
467
+ test('move multiple import statements with alias', () => {
468
+ const code = `
469
+ // @filename: test1.py
470
+ //// import json as j
471
+ //// import json.encoder as j2
472
+ ////
473
+ //// [|/*src*/a = j2.JSONEncoder()
474
+ //// b = j.loads("")|]
475
+
476
+ // @filename: test2.py
477
+ //// [|/*dest*/{|"r":"import json as j!n!import json.encoder as j2!n!!n!!n!"|}|]
478
+ ////
479
+ `;
480
+
481
+ testImportMove(code);
482
+ });
483
+
484
+ test('move multiple import statements - nested', () => {
485
+ const code = `
486
+ // @filename: test1.py
487
+ //// def foo():
488
+ //// import json
489
+ //// import json.encoder
490
+ ////
491
+ //// [|/*src*/a = json.encoder.JSONEncoder()
492
+ //// b = json.loads("")|]
493
+
494
+ // @filename: test2.py
495
+ //// [|/*dest*/{|"r":"import json!n!import json.encoder!n!!n!!n!"|}|]
496
+ ////
497
+ `;
498
+
499
+ testImportMove(code);
500
+ });
501
+
502
+ test('move multiple import statements - part of nested body', () => {
503
+ const code = `
504
+ // @filename: test1.py
505
+ //// def foo():
506
+ //// import json
507
+ //// import json.encoder
508
+ ////
509
+ //// [|/*src*/a = json.encoder.JSONEncoder()|]
510
+ //// b = json.loads("")
511
+
512
+ // @filename: test2.py
513
+ //// [|/*dest*/{|"r":"import json.encoder!n!!n!!n!"|}|]
514
+ ////
515
+ `;
516
+
517
+ testImportMove(code);
518
+ });
519
+
520
+ test('move multiple import statements - multi dotted name', () => {
521
+ const code = `
522
+ // @filename: nested/__init__.py
523
+ //// def foo(): pass
524
+
525
+ // @filename: nested/nested2/__init__.py
526
+ //// def foo(): pass
527
+
528
+ // @filename: nested/nested2/module.py
529
+ //// def foo(): pass
530
+
531
+ // @filename: test1.py
532
+ //// import nested
533
+ //// import nested.nested2
534
+ //// import nested.nested2.module
535
+ ////
536
+ //// nested.foo()
537
+ ////
538
+ //// [|/*src*/nested.nested2.foo()
539
+ //// nested.nested2.module.foo()|]
540
+
541
+ // @filename: test2.py
542
+ //// [|/*dest*/{|"r":"import nested.nested2!n!import nested.nested2.module!n!!n!!n!"|}|]
543
+ `;
544
+
545
+ testImportMove(code);
546
+ });
547
+
548
+ test('move simple from import statement', () => {
549
+ const code = `
550
+ // @filename: test1.py
551
+ //// from typing import Any
552
+ ////
553
+ //// [|/*src*/a: Any = 1|]
554
+
555
+ // @filename: test2.py
556
+ //// [|/*dest*/{|"r":"from typing import Any!n!!n!!n!"|}|]
557
+ ////
558
+ `;
559
+
560
+ testImportMove(code);
561
+ });
562
+
563
+ test('move from import statement with alias', () => {
564
+ const code = `
565
+ // @filename: test1.py
566
+ //// from typing import Any as t
567
+ ////
568
+ //// [|/*src*/a: t = 1|]
569
+
570
+ // @filename: test2.py
571
+ //// [|/*dest*/{|"r":"from typing import Any as t!n!!n!!n!"|}|]
572
+ ////
573
+ `;
574
+
575
+ testImportMove(code);
576
+ });
577
+
578
+ test('move submodule from import statement', () => {
579
+ const code = `
580
+ // @filename: test1.py
581
+ //// from json import encoder
582
+ ////
583
+ //// [|/*src*/a = encoder.JSONEncoder()|]
584
+
585
+ // @filename: test2.py
586
+ //// [|/*dest*/{|"r":"from json import encoder!n!!n!!n!"|}|]
587
+ ////
588
+ `;
589
+
590
+ testImportMove(code);
591
+ });
592
+
593
+ test('move submodule from import statement with alias', () => {
594
+ const code = `
595
+ // @filename: test1.py
596
+ //// from json import encoder as e
597
+ ////
598
+ //// [|/*src*/a = e.JSONEncoder()|]
599
+
600
+ // @filename: test2.py
601
+ //// [|/*dest*/{|"r":"from json import encoder as e!n!!n!!n!"|}|]
602
+ ////
603
+ `;
604
+
605
+ testImportMove(code);
606
+ });
607
+
608
+ test('move dotted from import statement', () => {
609
+ const code = `
610
+ // @filename: test1.py
611
+ //// from json.encoder import JSONEncoder
612
+ ////
613
+ //// [|/*src*/a = JSONEncoder()|]
614
+
615
+ // @filename: test2.py
616
+ //// [|/*dest*/{|"r":"from json.encoder import JSONEncoder!n!!n!!n!"|}|]
617
+ ////
618
+ `;
619
+
620
+ testImportMove(code);
621
+ });
622
+
623
+ test('move dotted from import statement with alias', () => {
624
+ const code = `
625
+ // @filename: test1.py
626
+ //// from json.encoder import JSONEncoder as j
627
+ ////
628
+ //// [|/*src*/a = j()|]
629
+
630
+ // @filename: test2.py
631
+ //// [|/*dest*/{|"r":"from json.encoder import JSONEncoder as j!n!!n!!n!"|}|]
632
+ ////
633
+ `;
634
+
635
+ testImportMove(code);
636
+ });
637
+
638
+ test('move both dotted from import and regular statement', () => {
639
+ const code = `
640
+ // @filename: test1.py
641
+ //// from json import loads
642
+ //// from json.encoder import JSONEncoder
643
+ ////
644
+ //// a = JSONEncoder()
645
+ //// [|/*src*/b = loads("")|]
646
+
647
+ // @filename: test2.py
648
+ //// [|/*dest*/{|"r":"from json import loads!n!!n!!n!"|}|]
649
+ ////
650
+ `;
651
+
652
+ testImportMove(code);
653
+ });
654
+
655
+ test('move both dotted from import and regular statement with alias', () => {
656
+ const code = `
657
+ // @filename: test1.py
658
+ //// from json import loads as j
659
+ //// from json.encoder import JSONEncoder
660
+ ////
661
+ //// a = JSONEncoder()
662
+ //// [|/*src*/b = j("")|]
663
+
664
+ // @filename: test2.py
665
+ //// [|/*dest*/{|"r":"from json import loads as j!n!!n!!n!"|}|]
666
+ ////
667
+ `;
668
+
669
+ testImportMove(code);
670
+ });
671
+
672
+ test('move multiple from import statements', () => {
673
+ const code = `
674
+ // @filename: test1.py
675
+ //// from json import loads
676
+ //// from json.encoder import JSONEncoder
677
+ ////
678
+ //// [|/*src*/a = JSONEncoder()
679
+ //// b = loads("")|]
680
+
681
+ // @filename: test2.py
682
+ //// [|/*dest*/{|"r":"from json import loads!n!from json.encoder import JSONEncoder!n!!n!!n!"|}|]
683
+ ////
684
+ `;
685
+
686
+ testImportMove(code);
687
+ });
688
+
689
+ test('move multiple from import statements with alias', () => {
690
+ const code = `
691
+ // @filename: test1.py
692
+ //// from json import loads as j
693
+ //// from json.encoder import JSONEncoder as j2
694
+ ////
695
+ //// [|/*src*/a = j2()
696
+ //// b = j("")|]
697
+
698
+ // @filename: test2.py
699
+ //// [|/*dest*/{|"r":"from json import loads as j!n!from json.encoder import JSONEncoder as j2!n!!n!!n!"|}|]
700
+ ////
701
+ `;
702
+
703
+ testImportMove(code);
704
+ });
705
+
706
+ test('move multiple from import statements - nested', () => {
707
+ const code = `
708
+ // @filename: test1.py
709
+ //// def foo():
710
+ //// from json import loads
711
+ //// from json.encoder import JSONEncoder
712
+ ////
713
+ //// [|/*src*/a = JSONEncoder()
714
+ //// b = loads("")|]
715
+
716
+ // @filename: test2.py
717
+ //// [|/*dest*/{|"r":"from json import loads!n!from json.encoder import JSONEncoder!n!!n!!n!"|}|]
718
+ ////
719
+ `;
720
+
721
+ testImportMove(code);
722
+ });
723
+
724
+ test('move multiple from import statements - part of nested body', () => {
725
+ const code = `
726
+ // @filename: test1.py
727
+ //// def foo():
728
+ //// from json import loads
729
+ //// from json.encoder import JSONEncoder
730
+ ////
731
+ //// [|/*src*/a = JSONEncoder()|]
732
+ //// b = loads("")
733
+
734
+ // @filename: test2.py
735
+ //// [|/*dest*/{|"r":"from json.encoder import JSONEncoder!n!!n!!n!"|}|]
736
+ ////
737
+ `;
738
+
739
+ testImportMove(code);
740
+ });
741
+
742
+ test('move multiple from import statements - multi dotted name', () => {
743
+ const code = `
744
+ // @filename: nested/__init__.py
745
+ //// def foo(): pass
746
+
747
+ // @filename: nested/nested2/__init__.py
748
+ //// def foo2(): pass
749
+
750
+ // @filename: nested/nested2/module.py
751
+ //// def foo3(): pass
752
+
753
+ // @filename: test1.py
754
+ //// from nested import foo
755
+ //// from nested.nested2 import foo2
756
+ //// from nested.nested2.module import foo3
757
+ ////
758
+ //// foo()
759
+ ////
760
+ //// [|/*src*/foo2()
761
+ //// foo3()|]
762
+
763
+ // @filename: test2.py
764
+ //// [|/*dest*/{|"r":"from nested.nested2 import foo2!n!from nested.nested2.module import foo3!n!!n!!n!"|}|]
765
+ `;
766
+
767
+ testImportMove(code);
768
+ });
769
+
770
+ test('relative path from import', () => {
771
+ const code = `
772
+ // @filename: nested/__init__.py
773
+ //// def foo(): pass
774
+
775
+ // @filename: nested/nested2/__init__.py
776
+ //// def foo2(): pass
777
+
778
+ // @filename: nested/nested2/module.py
779
+ //// def foo3(): pass
780
+
781
+ // @filename: nested/nested2/test1.py
782
+ //// from ...nested import foo
783
+ //// from ..nested2 import foo2
784
+ //// from .module import foo3
785
+ //// [|/*src*/foo()
786
+ //// foo2()
787
+ //// foo3()|]
788
+
789
+ // @filename: test2.py
790
+ //// [|/*dest*/{|"r":"from nested import foo!n!from nested.nested2 import foo2!n!from nested.nested2.module import foo3!n!!n!!n!"|}|]
791
+ `;
792
+
793
+ testImportMove(code);
794
+ });
795
+
796
+ test('namespace package from import', () => {
797
+ const code = `
798
+ // @filename: nested/module.py
799
+ //// def foo(): pass
800
+
801
+ // @filename: test1.py
802
+ //// from nested.module import foo
803
+ ////
804
+ //// [|/*src*/foo()|]
805
+
806
+ // @filename: test2.py
807
+ //// [|/*dest*/{|"r":"from nested.module import foo!n!!n!!n!"|}|]
808
+ `;
809
+
810
+ testImportMove(code);
811
+ });
812
+
813
+ test('namespace package with submodule from import', () => {
814
+ const code = `
815
+ // @filename: nested/module.py
816
+ //// def foo(): pass
817
+
818
+ // @filename: test1.py
819
+ //// from nested import module
820
+ ////
821
+ //// [|/*src*/module.foo()|]
822
+
823
+ // @filename: test2.py
824
+ //// [|/*dest*/{|"r":"from nested import module!n!!n!!n!"|}|]
825
+ `;
826
+
827
+ testImportMove(code);
828
+ });
829
+
830
+ test('multi nested namespace package with submodule from import', () => {
831
+ const code = `
832
+ // @filename: nested/nested2/nested3/module.py
833
+ //// def foo(): pass
834
+
835
+ // @filename: test1.py
836
+ //// from nested.nested2.nested3 import module
837
+ ////
838
+ //// [|/*src*/module.foo()|]
839
+
840
+ // @filename: test2.py
841
+ //// [|/*dest*/{|"r":"from nested.nested2.nested3 import module!n!!n!!n!"|}|]
842
+ `;
843
+
844
+ testImportMove(code);
845
+ });
846
+
847
+ test('multi nested namespace package with __init__ from import', () => {
848
+ const code = `
849
+ // @filename: nested/nested2/nested3/__init__.py
850
+ //// def foo(): pass
851
+
852
+ // @filename: test1.py
853
+ //// from nested.nested2.nested3 import foo
854
+ ////
855
+ //// [|/*src*/foo()|]
856
+
857
+ // @filename: test2.py
858
+ //// [|/*dest*/{|"r":"from nested.nested2.nested3 import foo!n!!n!!n!"|}|]
859
+ `;
860
+
861
+ testImportMove(code);
862
+ });
863
+
864
+ test('namespace package with relative path to root - from import', () => {
865
+ const code = `
866
+ // @filename: module.py
867
+ //// def foo(): pass
868
+
869
+ // @filename: test1.py
870
+ //// from . import module
871
+ ////
872
+ //// [|/*src*/module.foo()|]
873
+
874
+ // @filename: test2.py
875
+ //// [|/*dest*/{|"r":"from . import module!n!!n!!n!"|}|]
876
+ `;
877
+
878
+ testImportMove(code);
879
+ });
880
+
881
+ test('namespace package with relative path from import', () => {
882
+ const code = `
883
+ // @filename: nested/module.py
884
+ //// def foo(): pass
885
+
886
+ // @filename: test1.py
887
+ //// from .nested import module
888
+ ////
889
+ //// [|/*src*/module.foo()|]
890
+
891
+ // @filename: test2.py
892
+ //// [|/*dest*/{|"r":"from nested import module!n!!n!!n!"|}|]
893
+ `;
894
+
895
+ testImportMove(code);
896
+ });
897
+
898
+ test('namespace package import', () => {
899
+ const code = `
900
+ // @filename: nested/module.py
901
+ //// def foo(): pass
902
+
903
+ // @filename: test1.py
904
+ //// import nested.module
905
+ ////
906
+ //// [|/*src*/nested.module.foo()|]
907
+
908
+ // @filename: test2.py
909
+ //// [|/*dest*/{|"r":"import nested.module!n!!n!!n!"|}|]
910
+ `;
911
+
912
+ testImportMove(code);
913
+ });
914
+
915
+ test('__init__ at root', () => {
916
+ const code = `
917
+ // @filename: __init__.py
918
+ //// def foo(): pass
919
+
920
+ // @filename: test1.py
921
+ //// from . import foo
922
+ ////
923
+ //// [|/*src*/foo()|]
924
+
925
+ // @filename: test2.py
926
+ //// [|/*dest*/{|"r":"from . import foo!n!!n!!n!"|}|]
927
+ `;
928
+
929
+ testImportMove(code);
930
+ });
931
+
932
+ test('__init__ at root to nested file', () => {
933
+ const code = `
934
+ // @filename: __init__.py
935
+ //// def foo(): pass
936
+
937
+ // @filename: test1.py
938
+ //// from . import foo
939
+ ////
940
+ //// [|/*src*/foo()|]
941
+
942
+ // @filename: nested/test2.py
943
+ //// [|/*dest*/{|"r":"from .. import foo!n!!n!!n!"|}|]
944
+ `;
945
+
946
+ testImportMove(code);
947
+ });
948
+
949
+ test('move wild card imports', () => {
950
+ const code = `
951
+ // @filename: module.py
952
+ //// def foo(): pass
953
+ //// __all__ = [ 'foo' ]
954
+
955
+ // @filename: test1.py
956
+ //// from module import *
957
+ ////
958
+ //// [|/*src*/foo()|]
959
+
960
+ // @filename: nested/test2.py
961
+ //// [|/*dest*/{|"r":"from module import foo!n!!n!!n!"|}|]
962
+ `;
963
+
964
+ testImportMove(code);
965
+ });
966
+
967
+ test('move wild card imports from __init__', () => {
968
+ const code = `
969
+ // @filename: nested/__init__.py
970
+ //// def foo(): pass
971
+ //// __all__ = [ 'foo' ]
972
+
973
+ // @filename: test1.py
974
+ //// from nested import *
975
+ ////
976
+ //// [|/*src*/foo()|]
977
+
978
+ // @filename: nested/test2.py
979
+ //// [|/*dest*/{|"r":"from nested import foo!n!!n!!n!"|}|]
980
+ `;
981
+
982
+ testImportMove(code);
983
+ });
984
+
985
+ test('merge with existing import', () => {
986
+ const code = `
987
+ // @filename: test1.py
988
+ //// val1 = 1
989
+ //// val2 = 2
990
+ //// [|/*src*/a = val2|]
991
+
992
+ // @filename: test2.py
993
+ //// from test1 import val1[|/*dest*/{|"r":", val2"|}|]
994
+ `;
995
+
996
+ testImportMove(code);
997
+ });
998
+
999
+ test('merge multiple symbols with existing import', () => {
1000
+ const code = `
1001
+ // @filename: test1.py
1002
+ //// val1 = 1
1003
+ //// val2 = 2
1004
+ //// val3 = 3
1005
+ //// [|/*src*/a = val2
1006
+ //// b = val3|]
1007
+
1008
+ // @filename: test2.py
1009
+ //// from test1 import val1[|/*dest*/{|"r":", val2, val3"|}|]
1010
+ `;
1011
+
1012
+ testImportMove(code);
1013
+ });
1014
+
1015
+ test('move with existing import with wild card', () => {
1016
+ const code = `
1017
+ // @filename: test1.py
1018
+ //// val1 = 1
1019
+ //// val2 = 2
1020
+ //// [|/*src*/a = val2|]
1021
+
1022
+ // @filename: test2.py
1023
+ //// from test1 import *[|/*dest*/{|"r":"!n!from test1 import val2"|}|]
1024
+ `;
1025
+
1026
+ testImportMove(code);
1027
+ });
1028
+
1029
+ test('merge multiple symbols with multiple existing import and wildcard', () => {
1030
+ const code = `
1031
+ // @filename: test1.py
1032
+ //// val1 = 1
1033
+ //// val2 = 2
1034
+ //// val3 = 3
1035
+ //// [|/*src*/a = val2
1036
+ //// b = val3|]
1037
+
1038
+ // @filename: test2.py
1039
+ //// from test1 import *
1040
+ //// from test1 import val1[|/*dest*/{|"r":", val2, val3"|}|]
1041
+ `;
1042
+
1043
+ testImportMove(code);
1044
+ });
1045
+
1046
+ test('merge multiple symbols with multiple existing import', () => {
1047
+ const code = `
1048
+ // @filename: test1.py
1049
+ //// val1 = 1
1050
+ //// val2 = 2
1051
+ //// val3 = 3
1052
+ //// [|/*src*/a = val2
1053
+ //// b = val3|]
1054
+
1055
+ // @filename: test2.py
1056
+ //// from test1 import val1[|{|"r":", val3"|}|]
1057
+ //// from test1 import val2[|/*dest*/|]
1058
+ `;
1059
+
1060
+ testImportMove(code);
1061
+ });
1062
+
1063
+ test('merge multiple symbols with multiple existing import with alias', () => {
1064
+ const code = `
1065
+ // @filename: test1.py
1066
+ //// val1 = 1
1067
+ //// val2 = 2
1068
+ //// val3 = 3
1069
+ //// [|/*src*/a = val2
1070
+ //// b = val3|]
1071
+
1072
+ // @filename: test2.py
1073
+ //// from test1 import val1[|{|"r":", val2, val3"|}|]
1074
+ //// from test1 import val2 as v[|/*dest*/|]
1075
+ `;
1076
+
1077
+ testImportMove(code);
1078
+ });
1079
+
1080
+ test('skip with existing import statement', () => {
1081
+ const code = `
1082
+ // @filename: module.py
1083
+ //// val1 = 1
1084
+ //// val2 = 3
1085
+
1086
+ // @filename: test1.py
1087
+ //// import module
1088
+ ////
1089
+ //// [|/*src*/a = module.val1|]
1090
+
1091
+ // @filename: test2.py
1092
+ //// import module
1093
+ //// module.val2[|/*dest*/|]
1094
+ `;
1095
+
1096
+ testImportMove(code);
1097
+ });
1098
+
1099
+ test('skip with existing import statement with alias', () => {
1100
+ const code = `
1101
+ // @filename: module.py
1102
+ //// val1 = 1
1103
+ //// val2 = 3
1104
+
1105
+ // @filename: test1.py
1106
+ //// import module as m
1107
+ ////
1108
+ //// [|/*src*/a = m.val1|]
1109
+
1110
+ // @filename: test2.py
1111
+ //// import module as m
1112
+ //// m.val2[|/*dest*/|]
1113
+ `;
1114
+
1115
+ testImportMove(code);
1116
+ });
1117
+
1118
+ test('merge with existing import statement with alias', () => {
1119
+ const code = `
1120
+ // @filename: module.py
1121
+ //// val1 = 1
1122
+ //// val2 = 3
1123
+
1124
+ // @filename: test1.py
1125
+ //// import module
1126
+ ////
1127
+ //// [|/*src*/a = module.val1|]
1128
+
1129
+ // @filename: test2.py
1130
+ //// import module as m[|{|"r":"!n!import module"|}|]
1131
+ //// m.val2[|/*dest*/|]
1132
+ `;
1133
+
1134
+ testImportMove(code);
1135
+ });
1136
+
1137
+ test('merge with existing import statement with alias 2', () => {
1138
+ const code = `
1139
+ // @filename: module.py
1140
+ //// val1 = 1
1141
+ //// val2 = 3
1142
+
1143
+ // @filename: test1.py
1144
+ //// import module as m
1145
+ ////
1146
+ //// [|/*src*/a = m.val1|]
1147
+
1148
+ // @filename: test2.py
1149
+ //// import module[|{|"r":"!n!import module as m"|}|]
1150
+ //// module.val2[|/*dest*/|]
1151
+ `;
1152
+
1153
+ testImportMove(code);
1154
+ });
1155
+
1156
+ test('mixed with submodule and import - duplicated import', () => {
1157
+ const code = `
1158
+ // @filename: module.py
1159
+ //// val1 = 1
1160
+ //// val2 = 3
1161
+
1162
+ // @filename: test1.py
1163
+ //// import module
1164
+ ////
1165
+ //// [|/*src*/a = module.val1|]
1166
+
1167
+ // @filename: test2.py
1168
+ //// [|{|"r":"import module!n!"|}|]from . import module
1169
+ //// module.val2[|/*dest*/|]
1170
+ `;
1171
+
1172
+ testImportMove(code);
1173
+ });
1174
+
1175
+ test('mixed with submodule and import - duplicated import 2', () => {
1176
+ const code = `
1177
+ // @filename: module.py
1178
+ //// val1 = 1
1179
+ //// val2 = 3
1180
+
1181
+ // @filename: test1.py
1182
+ //// from . import module
1183
+ ////
1184
+ //// [|/*src*/a = module.val1|]
1185
+
1186
+ // @filename: test2.py
1187
+ //// [|{|"r":"from . import module!n!"|}|]import module
1188
+ //// module.val2[|/*dest*/|]
1189
+ `;
1190
+
1191
+ testImportMove(code);
1192
+ });
1193
+
1194
+ test('multiple mixed import statements', () => {
1195
+ const code = `
1196
+ // @filename: test1.py
1197
+ //// import typing
1198
+ //// from os import path
1199
+ //// import json.encoder as j
1200
+ //// import json.decoder
1201
+ ////
1202
+ //// [|/*src*/def foo(p1: str, p2: typing.Any, p3: typing.Union[int, str]):
1203
+ //// b = path.join(p1)
1204
+ //// e = j.JSONEncoder(skipkeys=True)
1205
+ //// d = json.decoder.JSONDecoder()|]
1206
+ // @filename: test2.py
1207
+ //// [|{|"r":"import json.decoder!n!import json.encoder as j!n!"|}|]import os[|{|"r":"!n!from os import path"|}|]
1208
+ //// import sys
1209
+ //// import typing
1210
+ //// from json import decoder[|/*dest*/|]
1211
+ `;
1212
+ testImportMove(code);
1213
+ });
1214
+
1215
+ test('multiple mixed import statements with merge', () => {
1216
+ const code = `
1217
+ // @filename: test1.py
1218
+ //// import typing
1219
+ //// from os import path
1220
+ //// from json import encoder as j
1221
+ //// from json import decoder
1222
+ ////
1223
+ //// [|/*src*/def foo(p1: str, p2: typing.Any, p3: typing.Union[int, str]):
1224
+ //// b = path.join(p1)
1225
+ //// e = j.JSONEncoder(skipkeys=True)
1226
+ //// d = decoder.JSONDecoder()|]
1227
+ // @filename: test2.py
1228
+ //// import sys
1229
+ //// import typing
1230
+ //// from os import abort[|{|"r":", path"|}|]
1231
+ //// from json import decoder[|{|"r":", encoder as j"|}|][|/*dest*/|]
1232
+ `;
1233
+ testImportMove(code);
1234
+ });
1235
+
1236
+ test('move into the same file import statement', () => {
1237
+ const code = `
1238
+ // @filename: test1.py
1239
+ //// import test2
1240
+ ////
1241
+ //// [|/*src*/test2.foo()|]
1242
+
1243
+ // @filename: test2.py
1244
+ //// [|{|"r":"import test2!n!!n!!n!"|}|]def foo(): pass
1245
+ //// [|/*dest*/|]
1246
+ `;
1247
+ testImportMove(code);
1248
+ });
1249
+
1250
+ test('move into the same file from import statement', () => {
1251
+ const code = `
1252
+ // @filename: test1.py
1253
+ //// from test2 import foo
1254
+ ////
1255
+ //// [|/*src*/foo()|]
1256
+
1257
+ // @filename: test2.py
1258
+ //// def foo(): pass
1259
+ //// [|/*dest*/|]
1260
+ `;
1261
+ testImportMove(code);
1262
+ });
1263
+
1264
+ test('move into the same file from import statement with alias', () => {
1265
+ const code = `
1266
+ // @filename: test1.py
1267
+ //// from test2 import foo as f
1268
+ ////
1269
+ //// [|/*src*/f()|]
1270
+
1271
+ // @filename: test2.py
1272
+ //// [|{|"r":"from test2 import foo as f!n!!n!!n!"|}|]def foo(): pass
1273
+ //// [|/*dest*/|]
1274
+ `;
1275
+ testImportMove(code);
1276
+ });
1277
+
1278
+ test('move into the same file from import statement for __init__', () => {
1279
+ const code = `
1280
+ // @filename: test1.py
1281
+ //// from nested import foo
1282
+ ////
1283
+ //// [|/*src*/foo()|]
1284
+
1285
+ // @filename: nested/__init__.py
1286
+ //// def foo(): pass
1287
+ //// [|/*dest*/|]
1288
+ `;
1289
+ testImportMove(code);
1290
+ });
1291
+
1292
+ test('move into the same file from import statement for __init__ with alias', () => {
1293
+ const code = `
1294
+ // @filename: test1.py
1295
+ //// from nested import foo as f
1296
+ ////
1297
+ //// [|/*src*/f()|]
1298
+
1299
+ // @filename: nested/__init__.py
1300
+ //// [|{|"r":"from nested import foo as f!n!!n!!n!"|}|]def foo(): pass
1301
+ //// [|/*dest*/|]
1302
+ `;
1303
+ testImportMove(code);
1304
+ });
1305
+
1306
+ test('move into the same file from import statement for submodule', () => {
1307
+ const code = `
1308
+ // @filename: test1.py
1309
+ //// from nested import module
1310
+ ////
1311
+ //// [|/*src*/module.foo()|]
1312
+
1313
+ // @filename: nested/__init__.py
1314
+ //// [|{|"r":"from nested import module!n!!n!!n!"|}|][|/*dest*/|]
1315
+
1316
+ // @filename: nested/module.py
1317
+ //// def foo(): pass
1318
+ `;
1319
+ testImportMove(code);
1320
+ });
1321
+
1322
+ function testImportMove(code: string) {
1323
+ const state = parseAndGetTestState(code).state;
1324
+
1325
+ const src = state.getRangeByMarkerName('src')!;
1326
+ const dest = state.getMarkerByName('dest');
1327
+
1328
+ const importMover = new ImportAdder(state.configOptions, state.importResolver, state.program.evaluator!);
1329
+ const importData = importMover.collectImportsForSymbolsUsed(
1330
+ state.program.getBoundSourceFile(src.fileName)!.getParseResults()!,
1331
+ TextRange.fromBounds(src.pos, src.end),
1332
+ CancellationToken.None
1333
+ );
1334
+
1335
+ const edits = importMover.applyImports(
1336
+ importData,
1337
+ state.program.getBoundSourceFile(dest.fileName)!.getParseResults()!,
1338
+ dest.position,
1339
+ CancellationToken.None
1340
+ );
1341
+
1342
+ assert(edits);
1343
+
1344
+ const ranges = state.getRanges().filter((r) => !!r.marker?.data);
1345
+ assert.strictEqual(edits.length, ranges.length);
1346
+
1347
+ for (const edit of edits) {
1348
+ assert(
1349
+ ranges.some((r) => {
1350
+ const data = r.marker!.data as { r: string };
1351
+ const expectedText = data.r;
1352
+ const expectedRange = state.convertPositionRange(r);
1353
+ return (
1354
+ rangesAreEqual(expectedRange, edit.range) &&
1355
+ expectedText.replace(/!n!/g, '\n') === edit.replacementText
1356
+ );
1357
+ }),
1358
+ `can't find '${edit.replacementText}'@'(${edit.range.start.line},${edit.range.start.character})'`
1359
+ );
1360
+ }
1361
+ }