def test_output_values(self, broadcaster, expected):
    """Test the broadcasted output value."""
    input_graph = utils_tf.data_dicts_to_graphs_tuple(
        [SMALL_GRAPH_1, SMALL_GRAPH_2])
    broadcasted = broadcaster(input_graph)
    with tf.Session() as sess:
      broadcasted_out = sess.run(broadcasted)
    self.assertNDArrayNear(
        np.array(expected, dtype=np.float32), broadcasted_out, err=1e-4)