The Python Apprentice

Counting lines

Let's break from the TDD rules and move a bit faster now. First we'll update the function to return the number of lines in the file:

# text_analyzer.pydef analyze_text(filename):    """Calculate the number of lines and characters in a file.    Args:      filename: The name of the file to analyze.    Raises:      IOError: If ``filename`` does not exist or can't be read.    Returns: The number of lines in the file.    """    with open(filename, 'r') as f:        return sum(1 for _ in f)

This change indeed gives us the results we want:

% python text_analyzer.py..----------------------------------------------------------------------Ran 2 tests in 0.003sOK