import sys

import pytest

from plangym.classic_control import ClassicControl
from tests.test_core import TestGymEnvironment


environments = [
    "MountainCar-v0",
    "Acrobot-v1",
    "MountainCarContinuous-v0",
    f"Pendulum-v0",
    "CartPole-v0",
]


@pytest.fixture(params=environments, scope="class")
def env(request) -> ClassicControl:
    return ClassicControl(name=request.param)
