import pytest
import solution
from solution import Solution


@pytest.mark.parametrize(
    "n,bad",
    [
        (5, 4),
        (1, 1),
        (10, 1),
        (10, 10),
    ],
)
def test_first_bad_version(n, bad, monkeypatch):
    monkeypatch.setattr(solution, "isBadVersion", lambda v: v >= bad)
    assert Solution().firstBadVersion(n) == bad