From 4878b85d2fdba9e7eaefc9ab8005d31d1f829f22 Mon Sep 17 00:00:00 2001 From: Patrick Elmer Date: Mon, 18 Sep 2023 10:33:38 +0900 Subject: [PATCH] Return empty string if not docstring exists --- magicli/magicli.py | 2 +- tests/test_truncate_docstring.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/magicli/magicli.py b/magicli/magicli.py index c1316f7..c7c3fe2 100644 --- a/magicli/magicli.py +++ b/magicli/magicli.py @@ -170,7 +170,7 @@ def make_lines(config): def truncate_docstring(docstring): if not docstring: - return docstring + return '' truncated_docstring = [] for line in docstring.split('\n'): if not line.lstrip(): diff --git a/tests/test_truncate_docstring.py b/tests/test_truncate_docstring.py index 82b1dbc..69c0f63 100644 --- a/tests/test_truncate_docstring.py +++ b/tests/test_truncate_docstring.py @@ -2,7 +2,7 @@ from magicli.magicli import truncate_docstring def test_no_docstring(): - assert truncate_docstring(None) == None + assert truncate_docstring(None) == '' def test_padded_multiline_string(): assert truncate_docstring("""This