magicli/tests/test_truncate_docstring.py

15 lines
442 B
Python

from magicli.magicli import truncate_docstring
def test_no_docstring():
assert truncate_docstring(None) == None
def test_padded_multiline_string():
assert truncate_docstring("""This
is
a docstring.""") == "This\nis\na docstring."
def test_remove_bottom():
assert truncate_docstring("""First line\n\nsecond line.""") == "First line"
assert truncate_docstring("""First line\n \nsecond line.""") == "First line"