def divide(a, b):
if b == 0:
raise ValueError('division by zero')
return a / b
import pytest
from source import divide
def test_divide_happy_path():
assert divide(10, 2) == 5
def test_divide_by_zero_raises():
with pytest.raises(ValueError):
divide(1, 0)