python/exercises/practice/proverb/proverb_test.py

import unittest

from proverb import (
    proverb,
)

# Tests adapted from `problem-specifications//canonical-data.json`
# PLEASE TAKE NOTE: Expected result lists for these test cases use **implicit line joining.**
# A new line in a result list below **does not** always equal a new list element.
# Check comma placement carefully!


class ProverbTest(unittest.TestCase):
    def test_zero_pieces(self):
        input_data = []
        self.assertEqual(proverb(*input_data, qualifier=None), [])

    def test_one_piece(self):
        input_data = ["nail"]
        self.assertEqual(
            proverb(*input_data, qualifier=None), ["And all for the want of a nail."]
        )

    def test_two_pieces(self):
        input_data = ["nail", "shoe"]
        self.assertEqual(
            proverb(*input_data, qualifier=None),
            [
                "For want of a nail the shoe was lost.",
                "And all for the want of a nail.",
            ],
        )

    def test_three_pieces(self):
        input_data = ["nail", "shoe", "horse"]
        self.assertEqual(
            proverb(*input_data, qualifier=None),
            [
                "For want of a nail the shoe was lost.",
                "For want of a shoe the horse was lost.",
                "And all for the want of a nail.",
            ],
        )

    def test_full_proverb(self):
        input_data = ["nail", "shoe", "horse", "rider", "message", "battle", "kingdom"]
        self.assertEqual(
            proverb(*input_data, qualifier=None),
            [
                "For want of a nail the shoe was lost.",
                "For want of a shoe the horse was lost.",
                "For want of a horse the rider was lost.",
                "For want of a rider the message was lost.",
                "For want of a message the battle was lost.",
                "For want of a battle the kingdom was lost.",
                "And all for the want of a nail.",
            ],
        )

    def test_four_pieces_modernized(self):
        input_data = ["pin", "gun", "soldier", "battle"]
        self.assertEqual(
            proverb(*input_data, qualifier=None),
            [
                "For want of a pin the gun was lost.",
                "For want of a gun the soldier was lost.",
                "For want of a soldier the battle was lost.",
                "And all for the want of a pin.",
            ],
        )

    # Track-specific tests

    def test_an_optional_qualifier_can_be_added(self):
        input_data = ["nail"]
        self.assertEqual(
            proverb(*input_data, qualifier="horseshoe"),
            ["And all for the want of a horseshoe nail."],
        )

    def test_an_optional_qualifier_in_the_final_consequences(self):
        input_data = ["nail", "shoe", "horse", "rider", "message", "battle", "kingdom"]
        self.assertEqual(
            proverb(*input_data, qualifier="horseshoe"),
            [
                "For want of a nail the shoe was lost.",
                "For want of a shoe the horse was lost.",
                "For want of a horse the rider was lost.",
                "For want of a rider the message was lost.",
                "For want of a message the battle was lost.",
                "For want of a battle the kingdom was lost.",
                "And all for the want of a horseshoe nail.",
            ],
        )