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