PY-16904 Preserve none indent when updating Numpy docstring

This commit is contained in:
Mikhail Golubev
2015-09-17 13:01:29 +03:00
parent a1a112b959
commit f1d56bc8cc
4 changed files with 21 additions and 1 deletions
@@ -299,7 +299,7 @@ public abstract class SectionBasedDocStringUpdater extends DocStringUpdater<Sect
protected String getFieldIndent(@NotNull Section section, @NotNull SectionField field) {
final String titleIndent = getSectionIndent(section);
final String fieldIndent = getLineIndent(getFieldStartLine(field));
final int diffSize = Math.max(1, PyIndentUtil.getLineIndentSize(fieldIndent) - PyIndentUtil.getLineIndentSize(titleIndent));
final int diffSize = Math.max(0, PyIndentUtil.getLineIndentSize(fieldIndent) - PyIndentUtil.getLineIndentSize(titleIndent));
return StringUtil.repeatSymbol(' ', diffSize);
}
@@ -0,0 +1,8 @@
def f(param1, param2):
"""
Parameters
----------
param1
param2 : object
"""
@@ -0,0 +1,7 @@
def <caret>f(param1, param2):
"""
Parameters
----------
param2 : object
"""
@@ -550,6 +550,11 @@ public class PyIntentionTest extends PyTestCase {
}
}
// PY-16904
public void testNumpyAddMissingParameterPreservesNoneIndent() {
doDocAddMissingParamsTest(DocStringFormat.NUMPY);
}
// PY-9795
public void testAddMissingParamsInGoogleDocString() {
doDocAddMissingParamsTest(DocStringFormat.GOOGLE);