code: def test_cleanup_dataset(tmpdir):
      """
      Test cleanup_dataset removes both data dir and archive
      """
      data_home = tmpdir.mkdir("fixtures")
  
      # Make dataset and archive
      foo = data_home.mkdir("foo")
      fdata = foo.join("foo.csv")
      fdata.write("testing 1,2,3")
  
      fzip = data_home.join("foo.zip")
      fzip.write("this is the archive file")
  
      # Make sure the files exist
      assert os.path.exists(fzip)
      assert os.path.exists(fdata)
  
      # Cleanup the dataset
      cleanup_dataset("foo", data_home=data_home)
  
      # Files should be gone
      assert not os.path.exists(fzip)
      assert not os.path.exists(fdata)
  
file_path: yellowbrick/tests/test_datasets/test_path.py
imports: import os
  import pytest
  import contextlib
  from yellowbrick.datasets.path import *
  from yellowbrick.exceptions import DatasetsError
label_desc: Use image node_id 10 for calling a global function with example usage: test_cleanup_dataset(tmpdir) without return types