add util tests

aerickson · Aug 7, 2024 · 78edb7a · 78edb7a
1 parent 55064c7
commit 78edb7a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tf_authoritative_scanner/util_test.py b/tf_authoritative_scanner/util_test.py
@@ -0,0 +1,9 @@
+import tf_authoritative_scanner.util as util
+
+
+def test_remove_leading_trailing_newline():
+    assert util.remove_leading_trailing_newline("") == ""
+    assert util.remove_leading_trailing_newline("a") == "a"
+    assert util.remove_leading_trailing_newline("a\n") == "a"
+    assert util.remove_leading_trailing_newline("\na") == "a"
+    assert util.remove_leading_trailing_newline("a\nb") == "a\nb"