def test_max_run_start():
"""Test max_run function for case where run is at the start of the
state"""
state = np.array([1, 1, 1, 1, 1, 1, 0, 2, 1, 1, 1, 1, 1, 4, 6, 1, 1])
assert max_run(1, state) == 6
def test_max_run_end():
"""Test max_run function for case where run is at the end of the
state"""
state = np.array([1, 1, 1, 1, 0, 1, 0, 2, 1, 1, 1, 1, 1, 1, 1, 1, 1])
assert max_run(1, state) == 9
def test_max_run_middle():
"""Test max_run function for case where run is in the middle of the
state"""
state = np.array([1, 1, 1, 1, 0, 1, 0, 2, 1, 1, 1, 1, 1, 4, 6, 1, 1])
assert max_run(1, state) == 5