Skip to content

Commit 6de1277

Browse files
committed
Added some tests
1 parent e135541 commit 6de1277

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

tests/test_visitors.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -208,6 +208,28 @@ def visit_Block(self, node):
208208
code,
209209
)
210210

211+
def test_DeleteBlock_keep(self):
212+
class DeleteBlock(jast.JNodeKeepTransformer):
213+
def visit_Block(self, node):
214+
return None
215+
216+
delete_block = DeleteBlock()
217+
new_tree = delete_block.visit(self.example)
218+
code = jast.unparse(new_tree)
219+
self.assertEqual(
220+
"public class Example {\n"
221+
" public int add(int a, int b);\n"
222+
" \n"
223+
" public static void main(String[] args);\n"
224+
"}",
225+
code,
226+
)
227+
self.assertIsNot(self.example, new_tree)
228+
self.assertEqual(
229+
self.source,
230+
jast.unparse(self.example),
231+
)
232+
211233
def test_DeleteReturn_keep(self):
212234
class DeleteReturn(jast.JNodeKeepTransformer):
213235
def visit_Return(self, node):

0 commit comments

Comments
 (0)