code: def test_fkpwl_distr_args_correct():
      x = mx.nd.array([[0.1, 0.5], [0.3, 0.99], [0.2, 0.6]])
  
      knot_spacings = mx.nd.array(
          [[0.1, 0.5, 0.4], [0.1, 0.5, 0.4], [0.1, 0.5, 0.4]]
      )
  
      fkpl_proj = FixedKnotsPiecewiseLinearOutput(
          [0.1, 0.6],
      ).get_args_proj()
  
      fkpl_proj.initialize()
      _, _, ks = fkpl_proj(x)
  
      assert np.allclose(ks.asnumpy(), knot_spacings.asnumpy())
  
file_path: gluonts/test/mx/distribution/test_piecewise_linear.py
imports: from typing import List, Tuple
  import mxnet
  import numpy
  import pytest
  from gluonts.core.serde import dump_json, load_json
  from gluonts.mx.distribution import FixedKnotsPiecewiseLinearOutput, PiecewiseLinear, PiecewiseLinearOutput
  from gluonts.testutil import empirical_cdf
label_desc: Use image node_id 6 for calling a global function with example usage: test_fkpwl_distr_args_correct() without return types