Repository: kedro-org/kedro
File: tests/framework/cli/test_starters.py (Python)

Code before:
    return "\n".join([project_name, repo_name, python_package])


def _make_cli_prompt_input_without_name(
    add_ons="none", repo_name="", python_package=""
):
    return "\n".join([add_ons, repo_name, python_package])


def _get_expected_files(add_ons: str, example_pipeline: str):
    add_ons_template_files = {
        "1": 0,  # Linting does not add any files
        "2": 3,  # If Testing is selected, we add 2 init.py files and 1 test_run.py
        "3": 1,  # If Logging is selected, we add logging.py
        "4": 2,  # If Documentation is selected, we add conf.py and index.rst
        "5": 8,  # If Data Structure is selected, we add 8 .gitkeep files
        "6": 2,  # If Pyspark is selected, we add spark.yml and hooks.py
        "7": 0,  # Kedro Viz does not add any files
    }  # files added to template by each add-on
    add_ons_list = _parse_add_ons_input(add_ons)
    example_pipeline_bool = _parse_yes_no_to_bool(example_pipeline)
    expected_files = FILES_IN_TEMPLATE_WITH_NO_ADD_ONS

    for add_on in add_ons_list:
        expected_files = expected_files + add_ons_template_files[add_on]
    # If example pipeline was choosen we don't need to delete /data folder
    if example_pipeline_bool and "5" not in add_ons_list:
        expected_files += add_ons_template_files["5"]
    example_files_count = [
        3,  # Raw data files
        2,  # Parameters_ .yml files
        6,  # .py files in pipelines folder
    ]
    if example_pipeline_bool:  # If example option is chosen
        expected_files += sum(example_files_count)
        expected_files += (
            4 if "7" in add_ons_list else 0
        )  # add 3 .py and 1 parameters files in reporting for Viz
        expected_files += (
            1 if "2" in add_ons_list else 0
        )  # add 1 test file if tests is chosen in add_ons

    return expected_files


def _assert_requirements_ok(
    result,
    add_ons="none",
    repo_name="new-kedro-project",
    output_dir=".",
):

Review comment (suggestion):
```suggestion
    # If example pipeline was chosen we don't need to delete /data folder
```

Code after:
    return "\n".join([project_name, repo_name, python_package])


def _make_cli_prompt_input_without_name(
    add_ons="none", repo_name="", python_package=""
):
    return "\n".join([add_ons, repo_name, python_package])


def _get_expected_files(add_ons: str, example_pipeline: str):
    add_ons_template_files = {
        "1": 0,  # Linting does not add any files
        "2": 3,  # If Testing is selected, we add 2 init.py files and 1 test_run.py
        "3": 1,  # If Logging is selected, we add logging.py
        "4": 2,  # If Documentation is selected, we add conf.py and index.rst
        "5": 8,  # If Data Structure is selected, we add 8 .gitkeep files
        "6": 2,  # If Pyspark is selected, we add spark.yml and hooks.py
        "7": 0,  # Kedro Viz does not add any files
    }  # files added to template by each add-on
    add_ons_list = _parse_add_ons_input(add_ons)
    example_pipeline_bool = _parse_yes_no_to_bool(example_pipeline)
    expected_files = FILES_IN_TEMPLATE_WITH_NO_ADD_ONS

    for add_on in add_ons_list:
        expected_files = expected_files + add_ons_template_files[add_on]
    # If example pipeline was chosen we don't need to delete /data folder
    if example_pipeline_bool and "5" not in add_ons_list:
        expected_files += add_ons_template_files["5"]
    example_files_count = [
        3,  # Raw data files
        2,  # Parameters_ .yml files
        6,  # .py files in pipelines folder
    ]
    if example_pipeline_bool:  # If example option is chosen
        expected_files += sum(example_files_count)
        expected_files += (
            4 if "7" in add_ons_list else 0
        )  # add 3 .py and 1 parameters files in reporting for Viz
        expected_files += (
            1 if "2" in add_ons_list else 0
        )  # add 1 test file if tests is chosen in add_ons

    return expected_files


def _assert_requirements_ok(
    result,
    add_ons="none",
    repo_name="new-kedro-project",
    output_dir=".",
):