code: def test_align_loss_args():
      class TestLossFwd(DummyLoss):
          def __init__(
              self, num_targets, num_agent_inputs, num_head_outputs
          ):
              super(TestLossFwd, self).__init__(
                  num_targets, num_agent_inputs, num_head_outputs
              )
  
          def loss_forward(
              self,
              F,
              head_output_2,
              head_output_1,
              agent_input_2,
              target_0,
              agent_input_1,
              param1,
              param2,
          ):
              pass
  
      assert align_loss_args(
          [1, 2, 3], [4, 5, 6, 7], [8, 9], TestLossFwd(3, 4, 2)
      ) == [3, 2, 6, 8, 5]
  
imports: import pytest
  import mxnet
  from mxnet import nd
  import numpy
  from rl_coach.architectures.mxnet_components.utils import *
label_desc: Use image node_id 7 for calling a global function with example usage: test_align_loss_args() without return types