code: def pyunit_drop():
      # Import data and collect number of columns and number of rows
      pros = h2o.import_file(
          pyunit_utils.locate("smalldata/prostate/prostate.csv")
      )
      nc = pros.ncol
      nr = pros.nrow
  
      # There are two ways to drop a single column: Pass in an int index or a string column,i.e., 1 or "C1"
      dropped_col_int = pros.drop(0)
      frame_checker(dropped_col_int)  # call on frame_checker()
      dropped_col_string = pros.drop("ID")
      frame_checker(dropped_col_string)  # call on frame_checker()
  
      # There are two ways to drop a set of columns: Pass in an int array or a string array, i.e., [1,2] or ["C1", "C2"]
      dropped_col_int_array = pros.drop([0, 1])
      frame_checker(dropped_col_int_array)  # call on frame_checker()
      to_drop = ["ID", "CAPSULE"]
      dropped_col_string_array = pros.drop(to_drop)
      frame_checker(dropped_col_string_array)  # call on frame_checker()
      assert to_drop == ["ID", "CAPSULE"]
  
      # Drop a first few rows from the frame (0 based)
      dropped_row_array_0 = pros.drop([0], axis=0)
      dropped_row_array_1 = pros.drop([0, 1], axis=0)
      dropped_row_array_2 = pros.drop([0, 1, 2], axis=0)
  
      # Drop last few rows from the frame (0 based)
      dropped_row_array_380 = pros.drop([379], axis=0)
      dropped_row_array_378 = pros.drop([378, 379], axis=0)
      dropped_row_array_377 = pros.drop([377, 378, 379], axis=0)
  
      # Check number of columns after drop are correct
      assert dropped_col_int.ncol == nc - 1
      assert dropped_col_string.ncol == nc - 1
      assert dropped_col_int_array.ncol == nc - 2
      assert dropped_col_string_array.ncol == nc - 2
  
      # #Check column names are correct after drop
      assert dropped_col_int.names == pros.names[1:]
      assert dropped_col_string.names == pros.names[1:]
      assert dropped_col_int_array.names == pros.names[2:]
      assert dropped_col_string_array.names == pros.names[2:]
  
      # #Check column types are correct after drop
      assert dropped_col_int.types == pros[1:].types
      assert dropped_col_string.types == pros[1:].types
      assert dropped_col_int_array.types == pros[2:].types
      assert dropped_col_string_array.types == pros[2:].types
  
      # Check number of rows after drop are correct
      assert dropped_row_array_0.nrow == nr - 1
      assert dropped_row_array_1.nrow == nr - 2
      assert dropped_row_array_2.nrow == nr - 3
      assert dropped_row_array_380.nrow == nr - 1
      assert dropped_row_array_378.nrow == nr - 2
      assert dropped_row_array_377.nrow == nr - 3
  
      # Check original frame preserves number of columns and rows
      assert pros.ncol == nc
      assert pros.nrow == nr
  
imports: import sys
  import h2o
  from tests import pyunit_utils
label_desc: Use image node_id 2 for calling a global function with example usage: pyunit_drop() without return types