Python和pytest測試

Question

我必須使用pytest做一些測試，但是我不知道從哪里開始。 這是我要測試的一段代碼：

def print_url_and_id():
    for item in movie_link[:100]:
        print item.contents[1], "The ID of this movie is:", '"' + item.contents[1]['href'][7:16] + '"'

誰能告訴我它的外觀如何？

Answer 1

您可以執行以下操作：

import pytest


def parametrized():
    expected_results = ["movie_link_01", "movie_link_02"]
    movie_link = ["movie_link_01", "movie_link_02", "movie_link_03"]
    # you can define your new_movie_link list like this as you have done, but instead of
    # printing it, add it to a this new_movie_link list
    return [(item_1, item_2) for item_1, item_2 in zip(movie_link[:2], expected_results)]


@pytest.mark.parametrize("movie_link, expected", parametrized())
def test_parametrizer(movie_link, expected):
    assert movie_link == expected