def test_format_docstring(self):
DOCSTRING = '''
a
b
c
'''
after_format = format_docstring(DOCSTRING)
expected_format = 'a b \n\n c'
self.assertEqual(after_format, expected_format)
def test_format_not_docstring(self):
after_format = format_docstring(None)
expected_format = ''
self.assertEqual(after_format, expected_format)