Python中pytest的参数化实例解析

1

2

3

4

5

6

7

8

import pytest

@pytest.mark.parametrize("x, y, expected", [

????(1, 2, 3),

????(3, 4, 7),

????(5, 6, 11),

])

def test_addition(x, y, expected):

????assert x + y == expected

1

2

3

4

5

6

7

8

import pytest

@pytest.mark.parametrize("test_input, expected_output", [

????([1, 2, 3], 6),

????([4, 5, 6], 15),

????([7, 8, 9], 24),

])

def test_sum(test_input, expected_output):

????assert sum(test_input) == expected_output

1

2

3

4

5

6

7

8

import pytest

@pytest.mark.parametrize("test_input, expected_output", [

????({"x": 1, "y": 2}, 3),

????({"x": 3, "y": 4}, 7),

????({"x": 5, "y": 6}, 11),

])

def test_addition(test_input, expected_output):

????assert test_input["x"] + test_input["y"] == expected_output

1

2

3

4

5

6

7

8

9

10

11

12

13

import pytest

import csv

def read_csv():

????with open('testdata.csv', 'r') as f:

????????reader = csv.reader(f)

????????rows = []

????????for row in reader:

????????????rows.append(row)

????????return rows[1:]

@pytest.mark.parametrize("test_input, expected_output", read_csv())

def test_addition(test_input, expected_output):

????x, y = map(int, test_input.split(','))

????assert x + y == int(expected_output)

1

2

3

4

5

6

7

8

9

10

11

12

13

14

import pytest

import time

def get_test_data():

????test_data = []

????start_time = time.time()

????while time.time() - start_time < 10:? # 运行时间小于 10 秒

????????x = random.randint(1, 100)

????????y = random.randint(1, 100)

????????expected = x + y

????????test_data.append((x, y, expected))

????return test_data

@pytest.mark.parametrize("x, y, expected", get_test_data())

def test_addition(x, y, expected):

????assert x + y == expected

1

2

3

4

5

6

7

8

9

10

11

12

13

14

import pytest

import requests

def get_test_data():

????response = requests.get('https://api.example.com/data')

????test_data = []

????for item in response.json():

????????x = item['x']

????????y = item['y']

????????expected = item['expected']

????????test_data.append((x, y, expected))

????return test_data

@pytest.mark.parametrize("x, y, expected", get_test_data())

def test_addition(x, y, expected):

????assert x + y == expected

1

2

3

4

5

import pytest

import itertools

@pytest.mark.parametrize("x, y", itertools.product([1, 2, 3], [4, 5, 6]))

def test_multiplication(x, y):

????assert x * y == y * x

1

2

3

4

5

6

7

8

9

10

11

import pytest

import random

def get_test_data():

????while True:

????????x = random.randint(1, 100)

????????y = random.randint(1, 100)

????????expected = x + y

????????yield (x, y, expected)

@pytest.mark.parametrize("x, y, expected", get_test_data())

def test_addition(x, y, expected):

????assert x + y == expected