code: def test_loss_output_dict():
      assert loss_output_dict([1, 2, 3], ["loss", "loss", "reg"]) == {
          "loss": [1, 2],
          "reg": [3],
      }
  
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 10 for calling a global function with example usage: test_loss_output_dict() without return types