Function create_random_schema

tests/flow/random_graph.py:226–258

Create random graph schema Example: res = create_random_schema(redis_graph, {"N": 100, "M": 200}, { "R": {"source": "N", "target": "M", "count": 500}})

()

def create_random_schema():
    """
    Create random graph schema
    Example:
    res = create_random_schema(redis_graph, {"N": 100, "M": 200}, {
                           "R": {"source": "N", "target": "M", "count": 500}})
    """
    nodes = []
    edges = []

    labels_count = randint(1, 10)
    for i in range(0, labels_count):
        count = randint(10, 100)
        properties = randint(0, 10)
        labels = [f"N{i}"]
        for j in range(0, randint(0, 5)):
            l = randint(1, labels_count)
            labels.append(f"N{l}")

        nodes.append(
            {"count": count, "properties": properties, "labels": labels})

    rel_types = randint(1, labels_count * 2)
    for i in range(0, rel_types):
        source_label = randint(0, labels_count - 1)
        source_count = nodes[source_label]["count"]
        target_label = randint(0, labels_count - 1)
        target_count = nodes[target_label]["count"]
        edge_count = randint(10, source_count * target_count)
        edges.append({"type": f"R{i}", "source": source_label,
                     "target": target_label, "count": edge_count})

    return nodes, edges


ALL_OPS = [create_node, create_edge, delete_node,