code: def test_identity(model_class):
      """
      test the Disrete environment vectorisation detection
  
      :param model_class: (BaseRLModel) the RL model
      """
      check_shape(lambda: IdentityEnv(dim=10), model_class, (), (1,))
  
file_path: stable-baselines/tests/test_auto_vec_detection.py
imports: import pytest
  import numpy
  from stable_baselines import A2C, ACER, ACKTR, DDPG, DQN, PPO1, PPO2, SAC, TRPO, TD3
  from stable_baselines.common.vec_env import DummyVecEnv
  from stable_baselines.common.identity_env import IdentityEnv, IdentityEnvBox, IdentityEnvMultiBinary, IdentityEnvMultiDiscrete
  from stable_baselines.common.evaluation import evaluate_policy
label_desc: Use image node_id 2 for calling a global function with example usage: test_identity(model_class) without return types