class PythonFileSummarizerTestCase(unittest.TestCase):
def setUp(self):
self.summarizer = PythonFileSummarizer()
def test_summarize_SimpleFunction(self):
code = """
def f1(a, b, c, d=None, **kwargs):
pass
"""
self._print_header(code)
self.summarizer.summarize(code)
def test_summarize_FunctionSpanMoreThanOneLine(self):
code = """
def f1(
a
):
pass
"""
self._print_header(code)
self.summarizer.summarize(code)
def test_summarize_ClassWithOneFunc(self):
code = """
class MyClass(object):
def f1(a, b, c, d=None, **kwargs):
pass
"""
self._print_header(code)
self.summarizer.summarize(code)
def _print_header(self, code_input):
print("{0:=^40}".format(" input "))
print(code_input)
print("{0:=^40}".format(" output "))
def setUp(self):
self.summarizer = PythonFileSummarizer()
