@zzzen/pyright-internal 1.1.254

package/dist/tests/indentationUtils.ptvs.test.js

@@ -0,0 +1,324 @@
+"use strict";
+/*
+ * indentationUtils.ptvs.test.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * Unit tests for indentationUtils module. some tests ported from ptvs indentation tests.
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert_1 = __importDefault(require("assert"));
+const indentationUtils_1 = require("../languageService/indentationUtils");
+const testState_1 = require("./harness/fourslash/testState");
+test('top level statement - pass', () => {
+    const code = `
+//// pass
+//// [|/*marker*/|]
+    `;
+    testIndentation(code, 0);
+});
+test('top level statement - function', () => {
+    const code = `
+//// def f():
+////  [|/*marker*/|]
+    `;
+    testIndentation(code, 4);
+});
+test('function with open paren at end of file', () => {
+    const code = `
+//// def f(
+//// [|/*marker*/|]
+    `;
+    // This is due to how our tokenizer associate new line at
+    // end of stream.
+    testIndentation(code, 0);
+});
+test('function with open paren between top level statement', () => {
+    const code = `
+//// def f(
+//// [|/*marker*/|]
+////
+//// def bar(): pass
+    `;
+    testIndentation(code, 4);
+});
+test('function with open paren', () => {
+    const code = `
+//// def f(
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('function with parameter', () => {
+    const code = `
+//// def f(x,
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('call with open paren at end of file', () => {
+    const code = `
+//// f(
+//// [|/*marker*/|]
+    `;
+    // This is due to how our tokenizer associate new line at
+    // end of stream.
+    testIndentation(code, 0);
+});
+test('call with open paren between top level statement', () => {
+    const code = `
+//// f(
+//// [|/*marker*/|]
+////
+//// bar()
+    `;
+    testIndentation(code, 4);
+});
+test('class with open paren', () => {
+    const code = `
+//// f(
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('call with parameter', () => {
+    const code = `
+//// f(x,
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('list', () => {
+    const code = `
+//// [
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('list with spaces', () => {
+    const code = `
+////          [
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 17);
+});
+test('list with nested', () => {
+    const code = `
+//// [[[[[[[
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('list with spaces and element', () => {
+    const code = `
+////          [x,
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 17);
+});
+test('list with nested with element', () => {
+    const code = `
+//// [[[[[[[x,
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('set', () => {
+    const code = `
+//// {
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('function body', () => {
+    const code = `
+//// def f():
+////     print('hi')
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('function body - pass', () => {
+    const code = `
+//// def f():
+////     pass
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 0);
+});
+test('list in dict', () => {
+    const code = `
+//// abc = {'x': [
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('nested list in dict', () => {
+    const code = `
+//// abc = {'x': [
+////     ['''str''',
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('nested list in dict', () => {
+    const code = `
+//// abc = {'x': [
+////     ['''str''',
+////      '''str2''']], 
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 5);
+});
+test('inner function', () => {
+    const code = `
+//// def f():
+////     print 'hi'
+////
+////     def inner(): pass
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('dict keys with comment', () => {
+    const code = `
+//// x = {  #comment
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 4);
+});
+test('dict first key with list', () => {
+    const code = `
+//// x = { #comment
+////       'a': [
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 6);
+});
+test('dict key list element on its own line', () => {
+    const code = `
+//// x = { #comment
+////       'a': [
+////           1,
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 10);
+});
+test('dict second key', () => {
+    const code = `
+//// x = { #comment
+////       'a': [
+////           1,
+////       ],
+//// [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 6);
+});
+test('after dict', () => {
+    const code = `
+//// x = { #comment
+////       'a': [
+////           1,
+////       ],
+////       'b': 42
+////     }
+////  [|/*marker*/|]
+////
+    `;
+    testIndentation(code, 0);
+});
+test('explicit multiline expression', () => {
+    const code = `
+//// def f():
+////     assert False, \\
+////       [|/*marker*/|]     
+////     'A message"
+////
+    `;
+    testIndentation(code, 8);
+});
+test('explicit multiline expression next statement', () => {
+    const code = `
+//// def f():
+////     assert False, \\
+////     'A message"
+////     [|/*marker*/|]     
+////
+    `;
+    testIndentation(code, 4);
+});
+test('nested block first', () => {
+    const code = `
+//// def a():
+////    [|/*marker*/|]
+////     if b():
+////         if c():
+////             d()
+////             p
+    `;
+    testIndentation(code, 4);
+});
+test('nested block second', () => {
+    const code = `
+//// def a():
+////     if b():
+////    [|/*marker*/|]
+////         if c():
+////             d()
+////             p
+    `;
+    testIndentation(code, 8);
+});
+test('nested block third', () => {
+    const code = `
+//// def a():
+////     if b():
+////         if c():
+////    [|/*marker*/|]
+////             d()
+////             p
+    `;
+    testIndentation(code, 12);
+});
+test('nested block last', () => {
+    const code = `
+//// def a():
+////     if b():
+////         if c():
+////             d()
+////    [|/*marker*/|]
+////             p
+    `;
+    testIndentation(code, 12);
+});
+function testIndentation(code, indentation, preferDedent) {
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const marker = state.getMarkerByName('marker');
+    const parseResults = state.program.getBoundSourceFile(marker.fileName).getParseResults();
+    const actual = (0, indentationUtils_1.getIndentation)(parseResults, marker.position, preferDedent);
+    assert_1.default.strictEqual(actual, indentation);
+}
+//# sourceMappingURL=indentationUtils.ptvs.test.js.map

package/dist/tests/indentationUtils.ptvs.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"indentationUtils.ptvs.test.js","sourceRoot":"","sources":["../../../src/tests/indentationUtils.ptvs.test.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;;;AAEH,oDAA4B;AAE5B,0EAAqE;AACrE,6DAAqE;AAErE,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE;IACpC,MAAM,IAAI,GAAG;;;KAGZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,gCAAgC,EAAE,GAAG,EAAE;IACxC,MAAM,IAAI,GAAG;;;KAGZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,yCAAyC,EAAE,GAAG,EAAE;IACjD,MAAM,IAAI,GAAG;;;KAGZ,CAAC;IAEF,yDAAyD;IACzD,iBAAiB;IACjB,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sDAAsD,EAAE,GAAG,EAAE;IAC9D,MAAM,IAAI,GAAG;;;;;KAKZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,0BAA0B,EAAE,GAAG,EAAE;IAClC,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACjC,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qCAAqC,EAAE,GAAG,EAAE;IAC7C,MAAM,IAAI,GAAG;;;KAGZ,CAAC;IAEF,yDAAyD;IACzD,iBAAiB;IACjB,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kDAAkD,EAAE,GAAG,EAAE;IAC1D,MAAM,IAAI,GAAG;;;;;KAKZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE;IAC/B,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE;IAC7B,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE;IACd,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAC1B,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;AAC9B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAC1B,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8BAA8B,EAAE,GAAG,EAAE;IACtC,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;AAC9B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+BAA+B,EAAE,GAAG,EAAE;IACvC,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,KAAK,EAAE,GAAG,EAAE;IACb,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE;IACvB,MAAM,IAAI,GAAG;;;;;KAKZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE;IAC9B,MAAM,IAAI,GAAG;;;;;KAKZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,cAAc,EAAE,GAAG,EAAE;IACtB,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE;IAC7B,MAAM,IAAI,GAAG;;;;;KAKZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE;IAC7B,MAAM,IAAI,GAAG;;;;;;KAMZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,gBAAgB,EAAE,GAAG,EAAE;IACxB,MAAM,IAAI,GAAG;;;;;;;KAOZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE;IAChC,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,0BAA0B,EAAE,GAAG,EAAE;IAClC,MAAM,IAAI,GAAG;;;;;KAKZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE;IAC/C,MAAM,IAAI,GAAG;;;;;;KAMZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;AAC9B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE;IACzB,MAAM,IAAI,GAAG;;;;;;;KAOZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,YAAY,EAAE,GAAG,EAAE;IACpB,MAAM,IAAI,GAAG;;;;;;;;;KASZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+BAA+B,EAAE,GAAG,EAAE;IACvC,MAAM,IAAI,GAAG;;;;;;KAMZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8CAA8C,EAAE,GAAG,EAAE;IACtD,MAAM,IAAI,GAAG;;;;;;KAMZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAC5B,MAAM,IAAI,GAAG;;;;;;;KAOZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE;IAC7B,MAAM,IAAI,GAAG;;;;;;;KAOZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;AAC7B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAC5B,MAAM,IAAI,GAAG;;;;;;;KAOZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;AAC9B,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE;IAC3B,MAAM,IAAI,GAAG;;;;;;;KAOZ,CAAC;IAEF,eAAe,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;AAC9B,CAAC,CAAC,CAAC;AAEH,SAAS,eAAe,CAAC,IAAY,EAAE,WAAmB,EAAE,YAAsB;IAC9E,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC,IAAI,CAAC,CAAC,KAAK,CAAC;IAC/C,MAAM,MAAM,GAAG,KAAK,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;IAE/C,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,kBAAkB,CAAC,MAAM,CAAC,QAAQ,CAAE,CAAC,eAAe,EAAG,CAAC;IAC3F,MAAM,MAAM,GAAG,IAAA,iCAAc,EAAC,YAAY,EAAE,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IAC3E,gBAAM,CAAC,WAAW,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;AAC5C,CAAC"}

package/dist/tests/indentationUtils.reindent.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/tests/indentationUtils.reindent.test.js

@@ -0,0 +1,332 @@
+"use strict";
+/*
+ * indentationUtils.test.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * Unit tests for indentationUtils module.
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert_1 = __importDefault(require("assert"));
+const textRange_1 = require("../common/textRange");
+const indentationUtils_1 = require("../languageService/indentationUtils");
+const testState_1 = require("./harness/fourslash/testState");
+test('re-indentation simple', () => {
+    const code = `
+//// [|/*marker*/def foo(): pass|]
+    `;
+    const expected = `def foo(): pass`;
+    testIndentation(code, 0, expected);
+});
+test('re-indentation indent first token', () => {
+    const code = `
+//// [|/*marker*/   def foo(): pass|]
+    `;
+    const expected = `  def foo(): pass`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation explicit multiline expression', () => {
+    const code = `
+//// [|/*marker*/def foo():
+////     i = \\
+////         1 \\
+////           + \\
+////         2|]
+    `;
+    const expected = `  def foo():
+      i = \\
+          1 \\
+            + \\
+          2`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation explicit multiline expression with multiple backslash', () => {
+    const code = `
+//// [|/*marker*/def foo():
+////     i = \\
+////         \\
+////         \\
+////         1|]
+    `;
+    const expected = `  def foo():
+      i = \\
+          \\
+          \\
+          1`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation multiline construct', () => {
+    const code = `
+//// [|/*marker*/def \\
+////     foo():
+////     i = 1 + 2|]
+    `;
+    const expected = `  def \\
+      foo():
+      i = 1 + 2`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation implicit multiline expression', () => {
+    const code = `
+//// [|/*marker*/def foo():
+////     i = (
+////          1
+////           +
+////          2
+////         )|]
+    `;
+    const expected = `  def foo():
+      i = (
+           1
+            +
+           2
+          )`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation implicit multiline expression with multiple blank lines', () => {
+    const code = `
+//// [|/*marker*/def foo():
+////     i = [
+////          1,
+////
+////          2
+////         ]|]
+    `;
+    const expected = `  def foo():
+      i = [
+           1,
+
+           2
+          ]`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation single line string', () => {
+    const code = `
+//// [|/*marker*/def foo():
+////     str = "string data"|]
+    `;
+    const expected = `  def foo():
+      str = "string data"`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation multiline line string', () => {
+    const code = `
+//// [|/*marker*/def foo():
+////     str = """first line
+////         second line
+////     last line
+//// """|]
+    `;
+    const expected = `  def foo():
+      str = """first line
+        second line
+    last line
+"""`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation with comments', () => {
+    const code = `
+//// [|/*marker*/def foo(): # comment at the end
+////     # commend above the line
+////     a = ( # commend at multiline expression
+////           1 + 2 # comment after expression
+////         )
+////     # commend before end of file|]
+    `;
+    const expected = `  def foo(): # comment at the end
+      # commend above the line
+      a = ( # commend at multiline expression
+            1 + 2 # comment after expression
+          )
+      # commend before end of file`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation with comments with backslash', () => {
+    const code = `
+//// [|/*marker*/def foo(): # comment at the end
+////     # commend above the line
+////     a = \\
+////         1 + 2 # comment after expression
+//// 
+////     # commend before end of file|]
+    `;
+    const expected = `  def foo(): # comment at the end
+      # commend above the line
+      a = \\
+          1 + 2 # comment after expression
+
+      # commend before end of file`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation doc comment', () => {
+    const code = `
+//// [|/*marker*/def foo():
+////     """ doc comment """
+////     a = 1|]
+    `;
+    const expected = `  def foo():
+      """ doc comment """
+      a = 1`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation multiline doc comment', () => {
+    const code = `
+//// [|/*marker*/def foo():
+////     """ doc comment 
+////         line 1
+////         line 2
+////     """
+////     a = 1|]
+    `;
+    const expected = `  def foo():
+      """ doc comment 
+          line 1
+          line 2
+      """
+      a = 1`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation top level multiline doc comment', () => {
+    const code = `
+//// [|/*marker*/    """ doc comment
+////         line 1
+////         line 2
+////     """
+////  a = 1|]
+    `;
+    const expected = `  """ doc comment
+      line 1
+      line 2
+  """
+a = 1`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation invalid code', () => {
+    const code = `
+//// [|/*marker*/ASDF
+//// ASDFASDFASD
+//// 
+//// asdf asdf asd fasdf sdf asdf asdf asdf
+////  23234235
+////    $%^#&*()_++
+//// asdfas asdf|]
+    `;
+    const expected = `  ASDF
+  ASDFASDFASD
+
+  asdf asdf asd fasdf sdf asdf asdf asdf
+   23234235
+     $%^#&*()_++
+  asdfas asdf`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation without first token', () => {
+    const code = `
+//// """ doc string """
+//// i = [|/*marker*/(
+////         1 + 2
+////     )|]
+    `;
+    const expected = `(
+    1 + 2
+)`;
+    testIndentation(code, 0, expected, /*indentFirstToken*/ false);
+});
+test('re-indentation single line variable doc string', () => {
+    const code = `
+//// [|/*marker*/i = 1
+//// """ doc string """|]
+    `;
+    const expected = `  i = 1
+  """ doc string """`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation multiple line variable doc string', () => {
+    const code = `
+//// [|/*marker*/i = 1
+//// """ doc string
+////     line 1
+//// """|]
+    `;
+    const expected = `  i = 1
+  """ doc string
+      line 1
+  """`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation single token', () => {
+    const code = `
+//// [|/*marker*/a|]
+    `;
+    const expected = `  a`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation between statements', () => {
+    const code = `
+//// def foo():
+////     pass
+//// [|/*marker*/i = 1|]
+//// """ doc string
+////     line 1
+//// """
+    `;
+    const expected = `  i = 1`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation inside of body', () => {
+    const code = `
+//// def foo():
+//// [|/*marker*/    """ doc string
+////         line 1
+////     """
+////     i = 10|]
+    `;
+    const expected = `  """ doc string
+      line 1
+  """
+  i = 10`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation tab', () => {
+    const code = `
+//// def foo():
+//// [|/*marker*/\t""" doc string
+//// \t\tline 1
+//// \t"""
+//// \ti = 10|]
+    `;
+    const expected = `  """ doc string
+\t  line 1
+  """
+  i = 10`;
+    testIndentation(code, 2, expected);
+});
+test('re-indentation tab on multiline text', () => {
+    const code = `
+//// def foo():
+//// [|/*marker*/\ta = """ line 1
+//// \t\tline 2
+//// \t"""
+//// \ti = 10
+//// \tif True:
+//// \t\tpass|]
+    `;
+    const expected = `  a = """ line 1
+\t\tline 2
+\t"""
+  i = 10
+  if True:
+\t  pass`;
+    testIndentation(code, 2, expected);
+});
+function testIndentation(code, indentation, expected, indentFirstToken = true) {
+    const state = (0, testState_1.parseAndGetTestState)(code).state;
+    const range = state.getRangeByMarkerName('marker');
+    const parseResults = state.program.getBoundSourceFile(range.fileName).getParseResults();
+    const actual = (0, indentationUtils_1.reindentSpan)(parseResults, textRange_1.TextRange.fromBounds(range.pos, range.end), indentation, indentFirstToken);
+    assert_1.default.strictEqual(actual, expected);
+}
+//# sourceMappingURL=indentationUtils.reindent.test.js.map

package/dist/tests/indentationUtils.reindent.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"indentationUtils.reindent.test.js","sourceRoot":"","sources":["../../../src/tests/indentationUtils.reindent.test.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;;;AAEH,oDAA4B;AAE5B,mDAAgD;AAChD,0EAAmE;AACnE,6DAAqE;AAErE,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE;IAC/B,MAAM,IAAI,GAAG;;KAEZ,CAAC;IAEF,MAAM,QAAQ,GAAG,iBAAiB,CAAC;IACnC,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE;IAC3C,MAAM,IAAI,GAAG;;KAEZ,CAAC;IAEF,MAAM,QAAQ,GAAG,mBAAmB,CAAC;IACrC,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8CAA8C,EAAE,GAAG,EAAE;IACtD,MAAM,IAAI,GAAG;;;;;;KAMZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;;YAIT,CAAC;IAET,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sEAAsE,EAAE,GAAG,EAAE;IAC9E,MAAM,IAAI,GAAG;;;;;;KAMZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;;YAIT,CAAC;IAET,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,oCAAoC,EAAE,GAAG,EAAE;IAC5C,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;gBAEL,CAAC;IAEb,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8CAA8C,EAAE,GAAG,EAAE;IACtD,MAAM,IAAI,GAAG;;;;;;;KAOZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;;;YAKT,CAAC;IAET,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,wEAAwE,EAAE,GAAG,EAAE;IAChF,MAAM,IAAI,GAAG;;;;;;;KAOZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;;;YAKT,CAAC;IAET,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE;IAC3C,MAAM,IAAI,GAAG;;;KAGZ,CAAC;IAEF,MAAM,QAAQ,GAAG;0BACK,CAAC;IAEvB,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sCAAsC,EAAE,GAAG,EAAE;IAC9C,MAAM,IAAI,GAAG;;;;;;KAMZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;;IAIjB,CAAC;IAED,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,8BAA8B,EAAE,GAAG,EAAE;IACtC,MAAM,IAAI,GAAG;;;;;;;KAOZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;;;mCAKc,CAAC;IAEhC,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,6CAA6C,EAAE,GAAG,EAAE;IACrD,MAAM,IAAI,GAAG;;;;;;;KAOZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;;;mCAKc,CAAC;IAEhC,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE;IACpC,MAAM,IAAI,GAAG;;;;KAIZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;YAET,CAAC;IAET,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sCAAsC,EAAE,GAAG,EAAE;IAC9C,MAAM,IAAI,GAAG;;;;;;;KAOZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;;;YAKT,CAAC;IAET,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,gDAAgD,EAAE,GAAG,EAAE;IACxD,MAAM,IAAI,GAAG;;;;;;KAMZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;;MAIf,CAAC;IAEH,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE;IACrC,MAAM,IAAI,GAAG;;;;;;;;KAQZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;;;;cAMP,CAAC;IAEX,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,oCAAoC,EAAE,GAAG,EAAE;IAC5C,MAAM,IAAI,GAAG;;;;;KAKZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;EAEnB,CAAC;IAEC,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,EAAE,oBAAoB,CAAC,KAAK,CAAC,CAAC;AACnE,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,gDAAgD,EAAE,GAAG,EAAE;IACxD,MAAM,IAAI,GAAG;;;KAGZ,CAAC;IAEF,MAAM,QAAQ,GAAG;qBACA,CAAC;IAElB,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,kDAAkD,EAAE,GAAG,EAAE;IAC1D,MAAM,IAAI,GAAG;;;;;KAKZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;MAGf,CAAC;IAEH,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE;IACrC,MAAM,IAAI,GAAG;;KAEZ,CAAC;IAEF,MAAM,QAAQ,GAAG,KAAK,CAAC;IAEvB,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE;IAC3C,MAAM,IAAI,GAAG;;;;;;;KAOZ,CAAC;IAEF,MAAM,QAAQ,GAAG,SAAS,CAAC;IAE3B,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,+BAA+B,EAAE,GAAG,EAAE;IACvC,MAAM,IAAI,GAAG;;;;;;KAMZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;SAGZ,CAAC;IAEN,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAC5B,MAAM,IAAI,GAAG;;;;;;KAMZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;SAGZ,CAAC;IAEN,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,sCAAsC,EAAE,GAAG,EAAE;IAC9C,MAAM,IAAI,GAAG;;;;;;;;KAQZ,CAAC;IAEF,MAAM,QAAQ,GAAG;;;;;SAKZ,CAAC;IAEN,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC,CAAC;AAEH,SAAS,eAAe,CAAC,IAAY,EAAE,WAAmB,EAAE,QAAgB,EAAE,gBAAgB,GAAG,IAAI;IACjG,MAAM,KAAK,GAAG,IAAA,gCAAoB,EAAC,IAAI,CAAC,CAAC,KAAK,CAAC;IAC/C,MAAM,KAAK,GAAG,KAAK,CAAC,oBAAoB,CAAC,QAAQ,CAAE,CAAC;IAEpD,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,kBAAkB,CAAC,KAAK,CAAC,QAAQ,CAAE,CAAC,eAAe,EAAG,CAAC;IAC1F,MAAM,MAAM,GAAG,IAAA,+BAAY,EACvB,YAAY,EACZ,qBAAS,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,GAAG,CAAC,EAC1C,WAAW,EACX,gBAAgB,CACnB,CAAC;IAEF,gBAAM,CAAC,WAAW,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;AACzC,CAAC"}

package/dist/tests/indentationUtils.test.d.ts

@@ -0,0 +1 @@
+export {};