test_fruit.py

import unittest

from fruit import capitalize_fruit_names


class TestFruit(unittest.TestCase):
    def test_empty_list(self):
        """with empty list"""
        self.assertEqual(capitalize_fruit_names([]), [])

    def test_lowercase(self):
        """with lowercase strings"""
        self.assertEqual(
            capitalize_fruit_names(["apple", "banana", "cherry"]),
            ["Apple", "Banana", "Cherry"],
        )

    def test_uppercase(self):
        """with uppercase strings"""
        self.assertEqual(
            capitalize_fruit_names(["APPLE", "BANANA", "CHERRY"]),
            ["Apple", "Banana", "Cherry"],
        )

    def test_mixed_case(self):
        """with mixed case strings"""
        self.assertEqual(
            capitalize_fruit_names(["mAnGo", "grApE"]),
            ["Mango", "Grape"],
        )

    def test_non_string_element(self):
        """with a mix of integer and string elements"""
        self.assertEqual(
            capitalize_fruit_names([123, "banana"]),
            ["", "Banana"],
        )


if __name__ == "__main__":
    unittest.main()