summaryrefslogtreecommitdiff
path: root/tests/unit/test_action_comments.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/test_action_comments.py')
-rw-r--r--tests/unit/test_action_comments.py47
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
+"""
+ )