code: def test_custom_keys(
      tmp_directory, nb_literals, nb_other_literals, arg, expected
  ):
      paths = ["nb_literals.ipynb", "nb_other_literals.ipynb"]
      col = NotebookCollection(paths, ids=arg)
      assert list(col.nbs) == expected
  
imports: import pytest
  import pandas
  import numpy
  from sklearn_evaluation import NotebookCollection
  from sklearn_evaluation.nb import NotebookCollection
  from IPython.display import HTML
label_desc: Use image node_id 2 for calling a global function with example usage: test_custom_keys(tmp_directory, nb_literals, nb_other_literals, arg, expected) without return types