Drill 18: Rewrite it — Testing with pytest

import pytest

def safe_divide(a, b):
    if b == 0:
        raise ValueError("Cannot divide by zero")
    return a / b

def test_safe_divide():
    assert safe_divide(10, 2) == 5

def test_zero_division():
    with pytest.raises(ValueError):
        safe_divide(1, 0)