code: def h2o_group_by_types():
      """
      This test checks that if the returned frame after a group_by operation returns correct type of group_by column.
      """
  
      data = h2o.H2OFrame(
          [
              ["4/1/07", 1, "A", 2.2],
              ["5/1/07", 23, "B", 223.4],
              ["6/1/07", 3, "A", 224.5],
          ],
          column_names=["date", "int", "string", "double"],
      )
  
      group_by_column = "date"
      grouped_type = get_group_by_type(data, group_by_column)
      assert (
          data[group_by_column].types == grouped_type
      ), "The type of group by column should be the same before and after group by."
  
      group_by_column = "int"
      grouped_type = get_group_by_type(data, group_by_column)
      assert (
          data[group_by_column].types == grouped_type
      ), "The type of group by column should be the same before and after group by."
  
      group_by_column = "double"
      grouped_type = get_group_by_type(data, group_by_column)
      assert (
          data[group_by_column].types == grouped_type
      ), "The type of group by column should be the same before and after group by."
  
imports: import sys
  import h2o
  from tests import pyunit_utils
label_desc: Use image node_id 1 for calling a global function with example usage: h2o_group_by_types() without return types