code: def test_args():
      def fun(x, f=lambda x: x, mul=1, add=0):
          return f(x) * mul + add
  
      # no args, single arg, tuple of args, dict of args
      p = (
          ggplot(data, aes("x"))
          + stat_function(fun=fun, size=2, color="blue")
          + stat_function(fun=fun, size=2, color="red", args=np.cos)
          + stat_function(
              fun=fun, size=2, color="green", args=(np.cos, 2, 1)
          )
          + stat_function(
              fun=fun,
              size=2,
              color="purple",
              args={"f": np.cos, "mul": 3, "add": 2},
          )
      )
  
      assert p == "args"
  
imports: import numpy
  import pandas
  import pytest
  from plotnine import aes, arrow, ggplot, stat_function
  from plotnine.exceptions import PlotnineError
label_desc: Use image node_id 2 for calling a global function with example usage: test_args() and returns: unknown