def test_lines_for(): build = Builder() extend_builder(build) for section in build.sections: if section != "1": assert build.lines_for(section) print section, build.lines_for(section) assert_equal( len(build.lines_for(section)), 2, "lines for section %s != 2" % (section))