def test_partition():
den = None
g = DGraph.read_dot(os.path.join(project_root(), "dot/weighted_g2.dot"))
print("testing partition on g2.dot for threshold=4:")
den = partition(g, SpectralCluster.SpectralCluster(), SizeCriteria(4))
assert len(den.node_list) == 10
print("testing partition on g2.dot for threshold=2:")
den = partition(g, SpectralCluster.SpectralCluster(), SizeCriteria(2))
print('expected:', len(den.node_list))
assert len(den.node_list) == 13
def test_labeling(self):
g = DGraph.read_dot(join(project_root(), "dot/large/ktails4.dot"))
den = partition.partition(g, SpectralCluster.SpectralCluster(),
stop_criteria.SizeCriteria(4))
labeler = PathLabeler.PathLabeler(g, den, labeling_on_type.EDGES)
labeler.label()
def test_labeling(self):
g = DGraph.read_dot(join(project_root(),
"dot/java.util.Formatter.dot"))
den = partition.partition(g, SpectralCluster.SpectralCluster(),
stop_criteria.SizeCriteria(4))
labeler = RandomWalkLabeler.RandomWalkLabeler(g, den, 'Both')
labeler.label()
def test_kerningham_lin(self):
g = DGraph.read_dot(
os.path.join(project_root(), "dot/cvs.net.mutated.dot"))
KL = KernighanLinCluster()
dendrogram = partition(g,
clustering_algo=KL,
stop_criterion=SizeCriteria(3))
print("foo")
def test_labeling(self):
g = DGraph.read_dot(os.path.join(project_root(), "dot/ssh.net.dot"))
den = partition.partition(g, SpectralCluster.SpectralCluster(), stop_criteria.SizeCriteria(4))
labeler = TfIdfLabeler.TfIdfLabeler(g, den, 'Both')
labeler.label()