code: def test_nasbench201():
      ss = searchspace.NasBench201()
      _test_searchspace_on_dataset(ss)
  
file_path: nni/test/algo/nas/hub/test_sanity.py
imports: from torchvision import transforms
  from torchvision.datasets import FakeData
  import pytest
  import pytorch_lightning
  import nni
  import nni.nas.evaluator.pytorch.lightning
  import nni.nas.hub.pytorch
  from nni.mutable import ConstraintViolation
  from nni.nas.nn.pytorch import ModelSpace
label_desc: Use image node_id 3 for calling a global function with example usage: test_nasbench201() without return types