18 lines
350 B
Python
18 lines
350 B
Python
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
|