def setup_class(cls): cls.NXGraph = nx.Graph edgelist = os.path.expandvars( "${GS_TEST_DIR}/dynamic/p2p-31_dynamic.edgelist") cls.g = nx.read_edgelist(edgelist, nodetype=int, data=True) cls.g.add_node(1, vdata_str="kdjfao") cls.g.add_node(1, vdata_int=123)
def setup_class(cls): cls.NXGraph = nx.DiGraph data_dir = os.path.expandvars("${GS_TEST_DIR}/ldbc_sample") cls.single_label_g = ldbc_sample_single_label(data_dir, True) cls.multi_label_g = ldbc_sample_multi_labels(data_dir, True) cls.p2p = load_p2p(os.path.expandvars("${GS_TEST_DIR}"), True) cls.p2p_nx = nx.read_edgelist( os.path.expandvars("${GS_TEST_DIR}/dynamic/p2p-31_dynamic.edgelist"), nodetype=int, data=True, create_using=nx.DiGraph, ) cls.str_oid_g = ldbc_sample_string_oid(data_dir, True)
def setup_class(cls): cls.grid_edges = [ (1, 2), (1, 5), (2, 3), (2, 6), (3, 4), (3, 7), (4, 8), (5, 6), (5, 9), (6, 7), (6, 10), (7, 8), (7, 11), (8, 12), (9, 10), (9, 13), (10, 11), (10, 14), (11, 12), (11, 15), (12, 16), (13, 14), (14, 15), (15, 16), ] cls.grid = nx.Graph() cls.grid.add_edges_from(cls.grid_edges, weight=1) cls.grid_ans = { 1: 0, 5: 1, 2: 1, 9: 2, 6: 2, 3: 2, 13: 3, 10: 3, 7: 3, 4: 3, 14: 4, 11: 4, 8: 4, 15: 5, 12: 5, 16: 6, } cls.grid_path_ans = { 3: 7, 4: 8, 8: 12, 12: 16, 1: 5, 5: 9, 9: 13, 2: 3, 6: 10, 10: 14, 7: 11, 11: 15, } data_dir = os.path.expandvars("${GS_TEST_DIR}") p2p_file = os.path.expandvars( "${GS_TEST_DIR}/dynamic/p2p-31_dynamic.edgelist") p2p_sub_file = os.path.expandvars( "${GS_TEST_DIR}/dynamic/p2p-31_dynamic_subgraph.edgelist") cls.p2p = nx.read_edgelist(p2p_file, nodetype=int, data=True, create_using=nx.DiGraph) cls.p2p_undirected = nx.read_edgelist(p2p_file, nodetype=int, data=True, create_using=nx.Graph) cls.p2p_subgraph = nx.read_edgelist(p2p_sub_file, nodetype=int, data=True, create_using=nx.DiGraph) cls.p2p_subgraph_undirected = nx.read_edgelist(p2p_sub_file, nodetype=int, data=True, create_using=nx.Graph) cls.p2p_length_ans = dict( pd.read_csv("{}/p2p-31-sssp".format(data_dir), sep=" ", header=None, prefix="").values) cls.p2p_dc_ans = dict( pd.read_csv( "{}/p2p-31-degree_centrality".format(data_dir), sep="\t", header=None, prefix="", ).values) cls.p2p_ev_ans = dict( pd.read_csv( "{}/p2p-31-eigenvector".format(data_dir), sep=" ", header=None, prefix="", ).values) cls.p2p_kz_ans = dict( pd.read_csv("{}/p2p-31-katz".format(data_dir), sep="\t", header=None, prefix="").values) cls.p2p_hits_ans = pd.read_csv( "{}/p2p-31-hits-directed".format(data_dir), sep="\t", header=None, prefix="") cls.p2p_pagerank_ans = dict( pd.read_csv( "{}/p2p-31-pagerank_nx".format(data_dir), sep="\t", header=None, prefix="", ).values) cls.p2p_clus_ans = dict( pd.read_csv("{}/p2p-31-clustering".format(data_dir), sep=" ", header=None, prefix="").values) cls.p2p_triangles_ans = dict( pd.read_csv("{}/p2p-31-triangles".format(data_dir), sep=" ", header=None, prefix="").values) cls.p2p_kcore_ans = sorted( pd.read_csv("{}/p2p-31-kcore".format(data_dir), sep=" ", header=None, prefix="").values) cls.empty_pagerank_ans = {}