erivlis / graphinate

    assert (parent_node_id, node_id) in builder_with_graph._graph.edges

def test_populate_nodes_callable_label_returns_non_string(builder_with_graph):

    node_type_absolute_id = ('parent', 'child')

    node = DummyNode(key='n5', value=123)

    def generator(**kwargs):

        yield node

    def label_fn(value):

        return 999  # Non-string label

    node_model = DummyNodeModel(generator=generator, label=label_fn)

    builder_with_graph.model._node_models = {node_type_absolute_id: [node_model]}

    builder_with_graph._populate_nodes(node_type_absolute_id)

    node_id = (node.key,)

    assert builder_with_graph._graph.nodes[node_id]['label'] == 999

def test_populate_nodes_empty_generator(builder_with_graph):

    assert len(builder_with_graph._graph.edges) == 0

def test_populate_nodes_callable_label_assignment(builder_with_graph):

    node_type_absolute_id = ('parent', 'child')

    node = DummyNode(key='n1', value=7)

    def generator(**kwargs):

        yield node

    def label_fn(value):

        return f"label-{value}"

    node_model = DummyNodeModel(generator=generator, label=label_fn)

    builder_with_graph.model._node_models = {node_type_absolute_id: [node_model]}

    builder_with_graph._populate_nodes(node_type_absolute_id)

    node_id = (node.key,)

    assert builder_with_graph._graph.nodes[node_id]['label'] == "label-7"

def test_populate_nodes_handles_invalid_parent_node_id(builder_with_graph):