test_project.py

import pytest
from unittest.mock import patch, mock_open
import csv
import random
from io import StringIO
from project import choose, save, randomize

@pytest.mark.parametrize("n, prompt_response, expected_output", [
    (1, "1", "option1"),
    (1, "2", "option2"),
    (2, "1 and 2", ("option1", "option2")),
])
@patch("project.Prompt.ask")
def test_choose(mock_prompt, n, prompt_response, expected_output):
    mock_prompt.return_value = prompt_response
    result = choose(n, "option1", "option2", "option3")
    assert result == expected_output

@patch("builtins.open", new_callable=mock_open)
@patch("csv.DictWriter")
def test_save(mock_dict_writer, mock_file):
    mock_writer_instance = mock_dict_writer.return_value
    inventory = ["item1", "item2"]
    with patch("project.Inventory", inventory):
        save("testfile", 10, 100, 50, "1", "1", 1)
    mock_file.assert_called_once_with("testfile.csv", "w")
    mock_writer_instance.writeheader.assert_called_once()
    mock_writer_instance.writerow.assert_called_once_with({
        "Level": 1,
        "Attack": 10,
        "Health": 100,
        "Defense": 50,
        "Defense Options": "1",
        "Attack Options": "1",
        "Inventory": inventory
    })

@pytest.mark.parametrize("number, expected_choices", [
    (1, ["Head", "Chest", "Legs"]),
    (2, ["Head", "Chest", "Legs"]),
])
@patch("random.choice")
@patch("random.sample")
def test_randomize(mock_sample, mock_choice, number, expected_choices):
    if number == 1:
        mock_choice.return_value = "Head"
        result = randomize(number)
        assert result == "Head"
        mock_choice.assert_called_once_with(expected_choices)
    elif number == 2:
        mock_sample.return_value = ("Head", "Chest")
        result = randomize(number)
        assert result == ("Head", "Chest")
        mock_sample.assert_called_once_with(expected_choices, k=2)