nimpretty: add #!nimpretty on/off directives

2 files changed, 20 insertions, 4 deletions

diff --git a/nimpretty/tests/exhaustive.nim b/nimpretty/tests/exhaustive.nim
index cfb42e7a9..ab58b7da4 100644
--- a/nimpretty/tests/exhaustive.nim
+++ b/nimpretty/tests/exhaustive.nim
@@ -315,9 +315,17 @@ proc f() =
     # escape char
     str &= c
 
-const test = r"C:\Users\-\Desktop\test.txt"
-
 proc getKeyAndData(cursor: int, op: int):
     tuple[key, data: string, success: bool] {.noInit.} =
   var keyVal: string
   var dataVal: string
+
+#!nimpretty off
+  when stuff:
+    echo "so nice"
+    echo "more"
+  else:
+     echo "misaligned"
+#!nimpretty on
+
+const test = r"C:\Users\-\Desktop\test.txt"
diff --git a/nimpretty/tests/expected/exhaustive.nim b/nimpretty/tests/expected/exhaustive.nim
index b9cb2e053..b114e9b8c 100644
--- a/nimpretty/tests/expected/exhaustive.nim
+++ b/nimpretty/tests/expected/exhaustive.nim
@@ -324,9 +324,17 @@ proc f() =
     # escape char
     str &= c
 
-const test = r"C:\Users\-\Desktop\test.txt"
-
 proc getKeyAndData(cursor: int; op: int):
     tuple[key, data: string; success: bool] {.noInit.} =
   var keyVal: string
   var dataVal: string
+
+#!nimpretty off
+  when stuff:
+    echo "so nice"
+    echo "more"
+  else:
+     echo "misaligned"
+#!nimpretty on
+
+const test = r"C:\Users\-\Desktop\test.txt"