def METHOD_NAME():
    """Tests to ensure that blackjack env step and reset functions return the correct types."""
    env = BlackjackFunctional()
    rng = jrng.PRNGKey(0)
    split_rng, rng = jrng.split(rng)
    state = env.initial(split_rng)
    env.action_space.seed(0)
    for t in range(10):
        obs = env.observation(state)
        action = env.action_space.sample()
        split_rng, rng = jrng.split(rng)
        next_state = env.transition(state, action, split_rng)
        reward = env.reward(state, action, next_state)
        terminal = env.terminal(next_state)
        assert len(state) == len(next_state)
        try:
            float(reward)
        except ValueError:
            pytest.fail("Reward is not castable to float")
        try:
            bool(terminal)
        except ValueError:
            pytest.fail("Terminal is not castable to bool")
        assert next_state[0].dtype == jnp.float32
        assert next_state[1].dtype == jnp.float32
        assert next_state[2].dtype == jnp.int32
        assert next_state[3].dtype == jnp.int32
        assert next_state[4].dtype == jnp.int32
        assert rng.dtype == jnp.uint32
        assert obs[0].dtype == jnp.int32
        assert obs[1].dtype == jnp.int32
        assert obs[2].dtype == jnp.int32
        state = next_state