diff --git a/tf_authoritative_scanner/util_test.py b/tf_authoritative_scanner/util_test.py new file mode 100644 index 0000000..8419ef8 --- /dev/null +++ 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"