code: def test_schema_row_count_cse() -> None:
      csv_a = NamedTemporaryFile()
      csv_a.write(
          b"""
      A,B
      Gr1,A
      Gr1,B
      """.strip()
      )
      csv_a.seek(0)
  
      df_a = pl.scan_csv(csv_a.name).with_row_count("Idx")
  
      result = (
          df_a.join(df_a, on="B")
          .group_by("A", maintain_order=True)
          .all()
          .collect(comm_subexpr_elim=True)
      )
  
      csv_a.close()
  
      expected = pl.DataFrame(
          {
              "A": ["Gr1"],
              "Idx": [[0, 1]],
              "B": [["A", "B"]],
              "Idx_right": [[0, 1]],
              "A_right": [["Gr1", "Gr1"]],
          },
          schema_overrides={
              "Idx": pl.List(pl.UInt32),
              "Idx_right": pl.List(pl.UInt32),
          },
      )
      assert_frame_equal(result, expected)
  
imports: import re
  from datetime import date, datetime
  from tempfile import NamedTemporaryFile
  from typing import Any
  import numpy
  import pytest
  import polars
  from polars.testing import assert_frame_equal
label_desc: Use image node_id 6 for calling a global function with example usage: test_schema_row_count_cse() without return types