code: def test_over_args() -> None:
      df = pl.DataFrame(
          {
              "a": ["a", "a", "b"],
              "b": [1, 2, 3],
              "c": [3, 2, 1],
          }
      )
  
      # Single input
      expected = pl.Series("c", [3, 3, 1]).to_frame()
      result = df.select(pl.col("c").max().over("a"))
      assert_frame_equal(result, expected)
  
      # Multiple input as list
      expected = pl.Series("c", [3, 2, 1]).to_frame()
      result = df.select(pl.col("c").max().over(["a", "b"]))
      assert_frame_equal(result, expected)
  
      # Multiple input as positional args
      result = df.select(pl.col("c").max().over("a", "b"))
      assert_frame_equal(result, expected)
  
imports: from __future__ import annotations
  import numpy
  import pytest
  import polars
  from polars.testing import assert_frame_equal, assert_series_equal
label_desc: Use image node_id 1 for calling a global function with example usage: test_over_args() without return types