diff options
Diffstat (limited to 'tests/unit/test_action_comments.py')
-rw-r--r-- | tests/unit/test_action_comments.py | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/tests/unit/test_action_comments.py b/tests/unit/test_action_comments.py new file mode 100644 index 00000000..508db0d2 --- /dev/null +++ b/tests/unit/test_action_comments.py @@ -0,0 +1,47 @@ +"""Tests for isort action comments, such as isort: skip""" +import isort + + +def test_isort_off_and_on(): + """Test so ensure isort: off action comment and associated on action comment work together""" + + # as top of file comment + assert ( + isort.code( + """# isort: off +import a +import a + +# isort: on +import a +import a +""" + ) + == """# isort: off +import a +import a + +# isort: on +import a +""" + ) + # as middle comment + assert ( + isort.code( + """ +import a +import a + +# isort: off +import a +import a +""" + ) + == """ +import a + +# isort: off +import a +import a +""" + ) |