code: def test_graph_branch(iotype, batch):
      odir = str(
          test_root_path
          / f"hls4mlprj_graph_branch_model_{iotype}_batch{batch}"
      )
      model = branch_model(odir, iotype)
      model.compile()
      hls4ml.utils.plot_model(
          model,
          show_shapes=True,
          show_precision=True,
          to_file=f"{odir}/model.png",
      )
      X0 = np.random.rand(batch, 1)
      X1 = np.random.rand(batch, 1)
      y_expected = 2 * (X0 + X1)
      y = model.predict([X0, X1]).reshape(y_expected.shape)
      # check the output
      np.testing.assert_allclose(y, y_expected, rtol=1, atol=2**-16)
  
file_path: hls4ml/test/pytest/test_graph.py
imports: from pathlib import Path
  import numpy
  import pytest
  import tensorflow
  import hls4ml
label_desc: Use image node_id 8 for calling a global function with example usage: test_graph_branch(iotype, batch) without return types