josephus_test.py

import pytest
from sol2 import solve


def brute_solve(n, step=2):
    curr = range(n)
    what = 0

    while len(curr) > 1:
        nwhat = (what - len(curr) % step) % step
        # nwhat = 1 - what if len(curr) % 2 == 1 else what
        curr = [val for i, val in enumerate(curr) if i % step != what]
        what = nwhat

    return curr[0]


@pytest.mark.parametrize("n", range(1, 201))
def test_it_works(n):
    assert solve(n) == brute_solve(n)
    print "solve(n): %s" % (solve(n),)


def test_speed():
    solve(12345678912)