github/OpenHands
1
0
Fork 0
mirror of https://github.com/OpenHands/OpenHands.git synced 2025-12-26 05:48:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Style/add return type hints (#5274)

Browse Source
This commit is contained in:
Cheng Yang 2024-11-27 00:42:45 +08:00 committed by GitHub
parent 0aa4a7184f
commit 71be744f2e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/runtime/conftest.py
View File

@ -29,16 +29,16 @@ project_dir = os.path.dirname(
sandbox_test_folder = '/openhands/workspace'
def _get_runtime_sid(runtime: Runtime):
def _get_runtime_sid(runtime: Runtime) -> str:
logger.debug(f'\nruntime.sid: {runtime.sid}')
return runtime.sid
def _get_host_folder(runtime: Runtime):
def _get_host_folder(runtime: Runtime) -> str:
return runtime.config.workspace_mount_path
def _get_sandbox_folder(runtime: Runtime):
def _get_sandbox_folder(runtime: Runtime) -> Path | None:
sid = _get_runtime_sid(runtime)
if sid:
return Path(os.path.join(sandbox_test_folder, sid))
@ -61,7 +61,7 @@ def _remove_folder(folder: str) -> bool:
return success
def _close_test_runtime(runtime: Runtime):
def _close_test_runtime(runtime: Runtime) -> None:
if isinstance(runtime, EventStreamRuntime):
runtime.close(rm_all_containers=False)
else:
@ -69,7 +69,7 @@ def _close_test_runtime(runtime: Runtime):
time.sleep(1)
def _reset_pwd():
def _reset_pwd() -> None:
global project_dir
# Try to change back to project directory
try:
@ -97,6 +97,7 @@ def print_method_name(request):
@pytest.fixture
def temp_dir(tmp_path_factory: TempPathFactory, request) -> str:
"""Creates a unique temporary directory.
Upon finalization, the temporary directory and its content is removed.
The cleanup function is also called upon KeyboardInterrupt.
@ -126,7 +127,7 @@ def temp_dir(tmp_path_factory: TempPathFactory, request) -> str:
# Depending on TEST_RUNTIME, feed the appropriate box class(es) to the test.
def get_runtime_classes():
def get_runtime_classes() -> list[type[Runtime]]:
runtime = TEST_RUNTIME
if runtime.lower() == 'eventstream':
return [EventStreamRuntime]
@ -138,7 +139,7 @@ def get_runtime_classes():
raise ValueError(f'Invalid runtime: {runtime}')
def get_run_as_openhands():
def get_run_as_openhands() -> list[bool]:
print(
'\n\n########################################################################'
)