tests/conftest.py

"""
Shared fixtures for the warehouse fulfillment test suite.
"""

import pytest
from grid_env.env import WarehouseFulfillmentEnv


@pytest.fixture()
def env_easy():
    env = WarehouseFulfillmentEnv(task_id="easy_single_pick", seed=7)
    env.reset()
    return env


@pytest.fixture()
def env_medium():
    env = WarehouseFulfillmentEnv(task_id="medium_multi_item", seed=7)
    env.reset()
    return env


@pytest.fixture()
def env_hard():
    env = WarehouseFulfillmentEnv(task_id="hard_restock_priority", seed=7)
    env.reset()
    return env


@pytest.fixture()
def env_obstacle_course():
    env = WarehouseFulfillmentEnv(task_id="obstacle_course", seed=7)
    env.reset()
    return env


@pytest.fixture()
def env_heavy_lifting():
    env = WarehouseFulfillmentEnv(task_id="heavy_lifting", seed=7)
    env.reset()
    return env


@pytest.fixture()
def env_stamina_run():
    env = WarehouseFulfillmentEnv(task_id="stamina_run", seed=7)
    env.reset()
    return env


@pytest.fixture()
def env_budget_run():
    env = WarehouseFulfillmentEnv(task_id="budget_run", seed=7)
    env.reset()
    return env


@pytest.fixture()
def env_gauntlet():
    env = WarehouseFulfillmentEnv(task_id="gauntlet", seed=7)
    env.reset()
    return env