def testFromEdgelist_ndarray(self):
name = "testFromEdgelist_tuple"
lst = np.array([[1, 2], [2, 2], [2, 3]])
d = convert.from_edgelist(name, lst, overide=True)
lst = np.array([[1, 2, 0.4], [2, 2, 2], [2, 3, 12]])
d = convert.from_edgelist(name, lst, overide=True)
edges = d.get_edges()
assert edges['src'].dtype == np.int
assert edges['dest'].dtype == np.int
assert edges['weight'].dtype == np.float32
def testFromEdgelist_dataframe(self):
name = "testFromEdgelist_dataframe"
lst = pd.DataFrame([[1, 2], [2, 2], [2, 3]], columns=['src', 'dest'])
d = convert.from_edgelist(name, lst, overide=True)
lst = pd.DataFrame(np.array([[1, 2, 0.4], [2, 2, 2], [2, 3, 12]]), columns=['src', 'dest', 'weight'])
d = convert.from_edgelist(name, lst, overide=True)
edges = d.get_edges()
assert edges['src'].dtype == np.int
assert edges['dest'].dtype == np.int
assert edges['weight'].dtype == np.float32
def testAsMirror(self):
name = sys._getframe().f_code.co_name
lst = [[1, 2], [2, 2], [2, 3], [2, 3], [3, 2]]
d = convert.from_edgelist(name, lst, directed=False, overide=True)
print (d.get_edges())
self.assertTrue(not d.is_edge_mirrored)
self.assertEqual(3, d.num_node)
self.assertEqual(3, d.num_edge)
edges = d.get_edges()
assert edges['src'].dtype == np.int
assert edges['dest'].dtype == np.int
self.assertTrue(not load_local(name).is_edge_mirrored)
newname = name + "_new"
d = d.as_mirror_edges(newname=newname, overide=True)
edges = d.get_edges()
print (d.get_edges())
self.assertTrue(d.is_edge_mirrored)
assert edges['src'].dtype == np.int
assert edges['dest'].dtype == np.int
self.assertEqual(3, d.num_node)
self.assertEqual(5, d.num_edge)
self.assertTrue(load_local(newname).is_edge_mirrored)
def load_sample_dataset(name, overide=False):
if not overide and local_exists(name):
return load_local(name)
else:
path = os.path.join(config.GCT_HOME, 'data', _DATASET_[name])
edges = pd.read_csv(path)
gt = pd.read_csv(path.replace("_edges", '_gt'))
description = ""
directed = False
return convert.from_edgelist(name,
edges,
groundtruth=gt,
directed=directed,
description=description,
overide=overide)
def testAsUndirect(self):
name = sys._getframe().f_code.co_name
lst = [[1, 2], [2, 2], [2, 3], [2, 3], [3, 2]]
d = convert.from_edgelist(name, lst, directed=True)
print (d.get_edges())
self.assertEqual(3, d.num_node)
self.assertEqual(4, d.num_edge)
edges = d.get_edges()
assert edges['src'].dtype == np.int
assert edges['dest'].dtype == np.int
d = d.as_undirected(newname=None)
edges = d.get_edges()
print (d.get_edges())
assert edges['src'].dtype == np.int
assert edges['dest'].dtype == np.int
self.assertEqual(3, d.num_node)
self.assertEqual(3, d.num_edge)