def anyGraphWithTransposedAdjListComputed(request): """ Create a Graph (or DiGraph) obj based on the param, compute the transposed adjacency list and return it. """ setFixtureParamNames(request, ["dataset", "managed_mem", "pool_allocator"]) csvFileName = request.param[0] reinitRMM(request.param[1], request.param[2]) G = createGraph(csvFileName) G.view_transposed_adj_list() return G
def graphWithAdjListComputed(request): """ Create a Graph obj from the CSV file in param, compute the adjacency list and return it. """ setFixtureParamNames(request, ["dataset", "managed_mem", "pool_allocator"]) csvFileName = request.param[0] reinitRMM(request.param[1], request.param[2]) G = createGraph(csvFileName, cugraph.structure.graph_classes.Graph) G.view_adj_list() return G
def edgelistCreated(request): """ Returns a new edgelist created from a CSV, which is specified as part of the parameterization for this fixture. """ # Since parameterized fixtures do not assign param names to param values, # manually call the helper to do so. Ensure the order of the name list # passed to it matches if there are >1 params. # If the request only contains n params, only the first n names are set. setFixtureParamNames(request, ["dataset", "managed_mem", "pool_allocator"]) csvFileName = request.param[0] reinitRMM(request.param[1], request.param[2]) return utils.read_csv_file(csvFileName)
def daskGraphFromDataset(request, client_connection): """ Returns a new dask dataframe created from the dataset file param. """ # Since parameterized fixtures do not assign param names to param values, # manually call the helper to do so. setFixtureParamNames(request, ["dataset"]) dataset = request.param chunksize = dcg.get_chunksize(dataset) ddf = dask_cudf.read_csv(dataset, chunksize=chunksize, delimiter=' ', names=['src', 'dst', 'value'], dtype=['int32', 'int32', 'float32']) dg = cugraph.DiGraph() dg.from_dask_cudf_edgelist(ddf, 'src', 'dst') return dg
def daskGraphFromDataset(request, dask_client): """ Returns a new dask dataframe created from the dataset file param. """ # Since parameterized fixtures do not assign param names to param values, # manually call the helper to do so. setFixtureParamNames(request, ["dataset"]) dataset = request.param chunksize = dcg.get_chunksize(dataset) ddf = dask_cudf.read_csv( dataset, chunksize=chunksize, delimiter=" ", names=["src", "dst", "value"], dtype=["int32", "int32", "float32"], ) dg = cugraph.DiGraph() dg.from_dask_cudf_edgelist(ddf, "src", "dst") return dg