class_docstr: Test visualization functions.
class_docstr_tok: ["Test","visualization","functions","."]
docstr: Test draw_boxes_side_by_side.
docstr_tok: ["Test","draw_boxes_side_by_side","."]
code: def test_draw_boxes_side_by_side(self):
      """Test draw_boxes_side_by_side."""
      pred, batch = self._generate_visualization_inputs()
      viz = detr_train_utils.draw_boxes_side_by_side(
          pred, batch, collections.defaultdict(lambda: "")
      )
      self.assertSequenceEqual(viz.shape, [4, 256, 512, 3])
  
file_path: scenic/scenic/projects/baselines/detr/tests/test_train_utils.py
imports: import collections
  import os
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  import jax.numpy
  import numpy
  from scenic.projects.baselines.detr import train_utils
label_desc: Use image node_id 3 for calling the DETRVisualizationsTest obj's underlying member method code with example usage: obj.test_draw_boxes_side_by_side() without return types