code: def check_wrong_return_type(self):
      # Test that we get a sensible exception invalid values passed to apply
      self._test_merge_error(
          fn=lambda l, r: l,
          output_schema=(
              StructType()
              .add("id", LongType())
              .add("v", ArrayType(YearMonthIntervalType()))
          ),
          error_class=NotImplementedError,
          error_message_regex="Invalid return type.*ArrayType.*YearMonthIntervalType",
      )
  
imports: import unittest
  from typing import cast
  from pyspark.sql.functions import array, explode, col, lit, udf, pandas_udf, sum
  from pyspark.sql.types import ArrayType, DoubleType, LongType, StructType, StructField, YearMonthIntervalType, Row
  from pyspark.sql.window import Window
  from pyspark.errors import IllegalArgumentException, PythonException
  from pyspark.testing.sqlutils import ReusedSQLTestCase, have_pandas, have_pyarrow, pandas_requirement_message, pyarrow_requirement_message
  from pyspark.testing.utils import QuietTest
label_desc: Use image node_id 31 for calling the CogroupedApplyInPandasTestsMixin obj's underlying member method code with example usage: obj.check_wrong_return_type() without return types