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