code: def pyunit_which():
      iris = h2o.import_file(
          pyunit_utils.locate("smalldata/iris/iris.csv")
      )
      setosa = (iris[4] == "Iris-setosa").which()
      versicolor = (iris[4] == "Iris-versicolor").which()
      virginica = (iris[4] == "Iris-virginica").which()
  
      assert sum(range(0, 50)) == setosa.sum()
      assert sum(range(50, 100)) == versicolor.sum()
      assert sum(range(100, 150)) == virginica.sum()
  
imports: from builtins import range
  import sys
  import h2o
  from tests import pyunit_utils
label_desc: Use image node_id 1 for calling a global function with example usage: pyunit_which() without return types