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()