erivlis / graphinate

    assert not np.array_equal(color_map['u1'], color_map['p1'])  # Different types, different colors

def test_node_color_mapping_filters_generic_node_type():

    # Arrange

    g = nx.Graph()

    # The 'node' type should be ignored in color calculations if other types exist

    g.graph['node_types'] = {'user': {}, 'post': {}, 'node': {}}

    g.add_node('u1', type='user')

    g.add_node('p1', type='post')

    g.add_node('n1', type='node')  # This should get a default color

    # Act

    color_map = node_color_mapping(g)

    # Assert

    # 'user' and 'post' should have distinct colors

    assert not np.array_equal(color_map['u1'], color_map['p1'])

    # 'node' type should map to the first color in the sequence (same as 'user' in this case)

    # because 'user' will be at index 0 and 'post' at index 1 after filtering.

    assert np.array_equal(color_map['n1'], color_map['u1'])

def test_node_color_mapping_with_only_generic_node_type():

    assert color_map == {}

def test_node_color_mapping_multiple_types():

    # Arrange

    g = nx.Graph()

    g.graph['node_types'] = {'user': {}, 'post': {}}

    g.add_node('u1', type='user')

    g.add_node('u2', type='user')

    g.add_node('p1', type='post')

    # Act

    color_map = node_color_mapping(g)

    # Assert

    assert len(color_map) == 3

    assert np.array_equal(color_map['u1'], color_map['u2'])  # Same type, same color

    assert not np.array_equal(color_map['u1'], color_map['p1'])  # Different types, different colors

def test_node_color_mapping_filters_generic_node_type():