def test_traceback_entry_getsource(self):

       tb = self.excinfo.traceback 

->     s = str(tb[-1].getsource() )

       assert s.startswith("def f():")

       assert s.endswith("raise ValueError") 

   def getsource(self): 

       """ return failing source code. """ 

       source = self.frame.code.fullsource

       start = self.getfirstlinesource()

       end = self.lineno

       try:

           _, end = source.getstatementrange(end) 

       except IndexError: 

           end = self.lineno + 1 

       # heuristic to stop displaying source on e.g. 

       #   if something:  # assume this causes a NameError

       #      # _this_ lines and the one 

              #        below we don't want from entry.getsource() 

       for i in range(self.lineno, end): 

           if source[i].rstrip().endswith(':'): 

               end = i + 1

               break 

->     return source[start:end]