code: def test_imdb(n_points):
      # test that fetch/download works fine
      X, y = shap.datasets.imdb(n_points=n_points)
  
      # check the shape of the result
      # check that the n_points parameter samples the dataset
      n_points = 25_000 if n_points is None else n_points
      assert len(X) == n_points
      assert len(y) == n_points
  
file_path: shap/tests/test_datasets.py
imports: import pytest
  import shap
label_desc: Use image node_id 4 for calling a global function with example usage: test_imdb(n_points) without return types