mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-20 15:34:23 +01:00
Feature/challenge memory management (#3425)
Co-authored-by: JS <38794445+jonathansheets517@users.noreply.github.com> Co-authored-by: Richard Beales <rich@richbeales.net>
This commit is contained in:
34
tests/integration/challenges/utils.py
Normal file
34
tests/integration/challenges/utils.py
Normal file
@@ -0,0 +1,34 @@
|
||||
from typing import Optional
|
||||
|
||||
import pytest
|
||||
|
||||
|
||||
def get_level_to_run(
|
||||
user_selected_level: Optional[int],
|
||||
level_currently_beaten: Optional[int],
|
||||
max_level: int,
|
||||
) -> int:
|
||||
"""
|
||||
Determines the appropriate level to run for a challenge, based on user-selected level, level currently beaten, and maximum level.
|
||||
|
||||
Args:
|
||||
user_selected_level (int | None): The level selected by the user. If not provided, the level currently beaten is used.
|
||||
level_currently_beaten (int | None): The highest level beaten so far. If not provided, the test will be skipped.
|
||||
max_level (int): The maximum level allowed for the challenge.
|
||||
|
||||
Returns:
|
||||
int: The level to run for the challenge.
|
||||
|
||||
Raises:
|
||||
ValueError: If the user-selected level is greater than the maximum level allowed.
|
||||
"""
|
||||
if user_selected_level is None:
|
||||
if level_currently_beaten is None:
|
||||
pytest.skip(
|
||||
"No one has beaten any levels so we cannot run the test in our pipeline"
|
||||
)
|
||||
# by default we run the level currently beaten.
|
||||
return level_currently_beaten
|
||||
if user_selected_level > max_level:
|
||||
raise ValueError(f"This challenge was not designed to go beyond {max_level}")
|
||||
return user_selected_level
|
||||
Reference in New Issue
Block a user