code: def cleanup(test_data, udfs, tmp_data, tmp_db):
      
      con = make_ibis_client(ENV)
  
      if udfs:
          # this comes before test_data bc the latter clobbers this too
          con.hdfs.rmdir(os.path.join(ENV.test_data_dir, 'udf'))
  
      if test_data:
          con.drop_database(ENV.test_data_db, force=True)
          con.hdfs.rmdir(ENV.test_data_dir)
  
      if tmp_data:
          con.hdfs.rmdir(ENV.tmp_dir)
  
      if tmp_db:
          con.drop_database(ENV.tmp_db, force=True)
docs: Cleanup Ibis test data and UDFs