from magicli.magicli import truncate_docstring


def test_no_docstring():
    assert truncate_docstring(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"