summaryrefslogtreecommitdiff
path: root/test/fixers/test_python_add_blank_lines_fixer.vader
blob: 7d042c8a9858ed87e211bc15a5f3f14788b7a67f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
Before:
  Save g:ale_fixers

After:
  Restore

Given python(Some Python without blank lines):
  def foo():
      """ This is a simple test docstring """
      return 1


  def bar():
      '''This is another simple test docstring'''
      return 1
      return 4


  def bar():
      """
      This is a multi-line
      docstring
      """

      if x:
          pass
          for l in x:
              pass
          for l in x:
              pass
              break
              continue
      elif x:
          pass
          while x:
              pass
          while x:
              pass
      else:
          pass
      if x:
          pass
      elif x:
          pass
      else:
          pass

Execute(Blank lines should be added appropriately):
  let g:ale_fixers = {'python': ['add_blank_lines_for_python_control_statements']}
  ALEFix

Expect python(Newlines should be added):
  def foo():
      """ This is a simple test docstring """

      return 1


  def bar():
      '''This is another simple test docstring'''

      return 1

      return 4


  def bar():
      """
      This is a multi-line
      docstring
      """

      if x:
          pass

          for l in x:
              pass

          for l in x:
              pass

              break

              continue
      elif x:
          pass

          while x:
              pass

          while x:
              pass
      else:
          pass

      if x:
          pass
      elif x:
          pass
      else:
          pass

Given python(A file with a main block):
  import os


  def main():
      print('hello')


  if __name__ == '__main__':
      main()

Execute(Fix the file):
  let g:ale_fixers = {'python': ['add_blank_lines_for_python_control_statements']}
  ALEFix

Expect python(extra newlines shouldn't be added to the main block):
  import os


  def main():
      print('hello')


  if __name__ == '__main__':
      main()


Given python(A file with variables/docstring that start with a control statement):
  def some():
      """
      This is a docstring that contains an
      break control statement and also contains a
      return something funny.
      """

      continue_some_var = True
      forward_something = False

      if (
          continue_some_var and
          forwarded_something
      ):
          return True


Execute(Fix the file):
  let g:ale_fixers = {'python': ['add_blank_lines_for_python_control_statements']}
  ALEFix

Expect python(Extra new lines are not added to the file):
  def some():
      """
      This is a docstring that contains an
      break control statement and also contains a
      return something funny.
      """

      continue_some_var = True
      forward_something = False

      if (
          continue_some_var and
          forwarded_something
      ):
          return True