TomsCodingCode
diff --git a/‎pre_commit_hooks/file_contents_sorter.py‎
Lines changed: 4 additions & 1 deletion b/‎pre_commit_hooks/file_contents_sorter.py‎
Lines changed: 4 additions & 1 deletion
diff --git a/‎tests/file_contents_sorter_test.py‎
Lines changed: 3 additions & 1 deletion b/‎tests/file_contents_sorter_test.py‎
Lines changed: 3 additions & 1 deletion
@@ -37,7 +37,10 @@ def sort_file_contents(
     after = sorted(lines, key=key)
 
     before_string = b''.join(before)
-    after_string = b'\n'.join(after) + b'\n'
+    after_string = b'\n'.join(after)
+
+    if after_string:
+        after_string += b'\n'
 
     if before_string == after_string:
         return PASS
 
@@ -10,7 +10,9 @@
 @pytest.mark.parametrize(
     ('input_s', 'argv', 'expected_retval', 'output'),
     (
-        (b'', [], FAIL, b'\n'),
+        (b'', [], PASS, b''),
+        (b'\n', [], FAIL, b''),
+        (b'\n\n', [], FAIL, b''),
         (b'lonesome\n', [], PASS, b'lonesome\n'),
         (b'missing_newline', [], FAIL, b'missing_newline\n'),
         (b'newline\nmissing', [], FAIL, b'missing\nnewline\n'),