def analyzeOneTree(i):
rt = RootedTree.random(30 * i)
rt.save(f"./experiment/tree_{30*i}_adjlist.txt")
roots = random.sample(rt.nodes, constant.NUM_VERTEX)
for root in roots:
newRT = RootedTree(rt.tree, root)
copmanager.getCopNumber(newRT)
treedrawer.drawRootedTree(
newRT,
showLabel=True,
savePath=f"./experiment/tree_{30*i}_labels_rootedat_{root}.pdf")
def example():
os.makedirs("./examples", exist_ok=True)
files = glob.glob('./examples/*')
for f in files:
os.remove(f)
rt = RootedTree.load(0, "example4_1.txt")
print(copmanager.getCopNumber(rt))
treedrawer.drawRootedTree(rt,
True,
show=False,
savePath="examples/example4.1.pdf")
rt = RootedTree.load(0, "example4_2.txt")
print(copmanager.getCopNumber(rt))
treedrawer.drawRootedTree(rt,
True,
show=False,
savePath="examples/example4.2.pdf")
rt = RootedTree.load(0, "example4_3.txt")
print(copmanager.getCopNumber(rt))
treedrawer.drawRootedTree(rt,
True,
show=False,
savePath="examples/example4.3.pdf")
rt = RootedTree.load(5, "example4_3.txt")
print(copmanager.getCopNumber(rt))
treedrawer.drawRootedTree(rt,
True,
show=False,
savePath="examples/example4.4.pdf")
rt = RootedTree.load(0, "example4_5.txt")
x1 = 11
x2 = 12
x3 = 13
rt.labels = {
0: None,
1: Label.make(4, constant.PERPEN_SYM, 0, 0, 1, 0),
2: Label.make(4, constant.PERPEN_SYM, 0, 0, 1, 0),
3: Label.make(5, x1, 0, 0, 0, 2).append(2, constant.PERPEN_SYM),
4: Label.make(8, x2, 0, 0, 0, 0).append(4, 4),
5: Label.make(6, x3, 0, 0, 0, 0).append(2, 5),
11: Label.make(5, 11),
12: Label.make(8, 12),
13: Label.make(6, 13)
}
print(copmanager.getCopNumber(rt))
treedrawer.drawRootedTree(rt,
True,
show=False,
savePath="examples/example4.5.pdf")
def test_example4_5(self):
rt = RootedTree.load(0, "example4_5.txt")
x1 = 11
x2 = 12
x3 = 13
rt.labels = {
0: None,
1: Label.make(4, constant.PERPEN_SYM, 0, 0, 1, 0),
2: Label.make(4, constant.PERPEN_SYM, 0, 0, 1, 0),
3: Label.make(5, x1, 0, 0, 0, 2).append(2, constant.PERPEN_SYM),
4: Label.make(8, x2, 0, 0, 0, 0).append(4, 4),
5: Label.make(6, x3, 0, 0, 0, 0).append(2, 5),
11: Label.make(5, 11),
12: Label.make(8, 12),
13: Label.make(6, 13)
}
self.assertEqual(copmanager.getCopNumber(rt),
Label.make(8, x2).append(7, constant.PERPEN_SYM))
def test_example4_4(self):
rt = RootedTree.load(5, "example4_3.txt")
self.assertEqual(copmanager.getCopNumber(rt),
Label.make(2, constant.PERPEN_SYM, 0, 0, 1, 0))