Pytest Fixtures and Tests

1import pytest
2 
3@pytest.fixture
4def sample_data():
5    return {
6        'users': ['alice', 'bob', 'charlie'],
7        'count': 3
8    }
9 
10@pytest.fixture
11def database():
12    db = create_connection()
13    yield db
14    db.close()
15 
16def test_user_count(sample_data):
17    assert len(sample_data['users']) == sample_data['count']
18 
19def test_user_names(sample_data):
20    assert 'alice' in sample_data['users']
21    assert 'dave' not in sample_data['users']