test_search.py

import pytest
from unittest.mock import patch
from mylib.search import final_result


@pytest.mark.parametrize(
    "query, expected_response",
    [
        (
            "walmart",
            [
                "OFFER: Arber, at Walmart\ncombined: ARBER WALMART",
                "OFFER: Back to the Roots Seeds, at Walmart\ncombined: Packaged Meals & Sides BACK TO THE ROOTS WALMART",
                "OFFER: OxiClean™ Laundry Stain Removers, select varieties at Walmart\ncombined: WALMART OXICLEAN",
                "OFFER: Gorton's at select retailers\ncombined: Frozen Seafood Jerky & Dried Meat Frozen Meals WALMART GORTONS",
                "OFFER: Gorton's Air Fried Butterfly Shrimp, at Walmart\ncombined: Frozen Seafood Jerky & Dried Meat Frozen Meals WALMART GORTONS",
                "OFFER: L’Oréal Paris Men Expert hair color, spend $9 at Walmart\ncombined: Hair Care LOREAL PARIS HAIR COLOR WALMART",
                "OFFER: L'Oréal Paris Men Expert hair color, spend $19 at Walmart\ncombined: Hair Care LOREAL PARIS HAIR COLOR WALMART",
                "OFFER: Back to the Roots Grow Kits at Walmart or The Home Depot\ncombined: Packaged Meals & Sides BACK TO THE ROOTS WALMART",
                "OFFER: Cooked Perfect® Meatballs, at Walmart\ncombined: Frozen Chicken COOKED PERFECT WALMART",
                "OFFER: Purex® laundry detergent, select varieties, at Walmart\ncombined: PUREX Laundry Supplies WALMART",
            ],
        ),
    ],
)
@patch("builtins.input", side_effect=["query_1", "query_2"])
def test_final_result(mock_input, query, expected_response):
    response = final_result(query)
    assert type(response) == type(expected_response)


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