def test_wide_layer_ids():
    graph = get_conv_dense_model()
    assert (len(graph.wide_layer_ids()) == 2)
def test_wide_layer_ids():
    graph = get_conv_dense_model()
    assert (len(graph.wide_layer_ids()) == 2)