def test_1(self):
ft = FractalTree(proportions=(1, 2, 3), tree_permutation_order=(3, 1, 2), value=10)
ft.add_layer()
ft.add_layer()
for node in ft.get_layer(1):
node.reduce_children(condition=lambda node: node.fractal_order == 1)
self.assertEqual([node.fractal_order for node in ft.traverse_leaves()], [2, 3, 3, 2, 2, 3])
def test_2(self):
ft = FractalTree(proportions=(1, 2, 3), tree_permutation_order=(3, 1, 2), value=10)
ft.add_layer()
ft.add_layer()
for node in ft.get_layer(1):
node.reduce_children(condition=lambda node: node.fractal_order == 1)
actual = [node.value for node in ft.traverse_leaves()]
expected = [Fraction(2, 1),
Fraction(3, 1),
Fraction(1, 1),
Fraction(2, 3),
Fraction(4, 3),
Fraction(2, 1)]
self.assertEqual(actual, expected)
class Test(TestCase):
def setUp(self) -> None:
self.ft = FractalTree(value=10)
def test_1(self):
self.ft.change_value(20)
result = 20
self.assertEqual(result, self.ft.value)
def test_2(self):
self.ft.add_layer()
self.ft.get_children()[0].change_value(10)
expected = 15
self.assertEqual(expected, self.ft.value)
def test_3(self):
self.ft.add_layer()
self.ft.get_children()[0].change_value(10)
expected = [Fraction(10, 1), Fraction(5, 3), Fraction(10, 3)]
self.assertEqual(expected,
[child.value for child in self.ft.get_children()])
def test_4(self):
self.ft.add_layer()
self.ft.change_value(15)
expected = [Fraction(15, 2), Fraction(5, 2), Fraction(5, 1)]
self.assertEqual(expected,
[child.value for child in self.ft.get_children()])
def test_5(self):
self.ft.add_layer()
self.ft.add_layer()
self.ft.get_children()[0].change_value(10)
expected = [[Fraction(15, 1)], [10,
Fraction(5, 3),
Fraction(10, 3)],
[[Fraction(5, 3),
Fraction(10, 3),
Fraction(5, 1)],
[Fraction(5, 6),
Fraction(5, 18),
Fraction(5, 9)],
[Fraction(10, 9),
Fraction(5, 3),
Fraction(5, 9)]]]
self.assertEqual(expected, [
self.ft.get_layer(layer=i, key='value')
for i in range(self.ft.number_of_layers + 1)
])
class Test(TestCase):
def setUp(self) -> None:
self.ft = FractalTree(proportions=[1, 2, 3],
tree_permutation_order=[3, 1, 2],
value=10)
def test_0(self):
with self.assertRaises(Exception):
self.ft.get_layer(1)
def test_1(self):
self.assertEqual([self.ft], self.ft.get_layer(0))
def test_2(self):
self.ft.add_layer()
result = self.ft.get_children()
self.assertEqual(result, self.ft.get_layer(1))
def test_3(self):
for i in range(3):
self.ft.add_layer()
result = self.ft.get_children()
self.assertEqual(result, self.ft.get_layer(1))
def test_4(self):
for i in range(3):
self.ft.add_layer()
result = [child.get_children() for child in self.ft.get_children()]
self.assertEqual(result, self.ft.get_layer(2))
def test_5(self):
for i in range(3):
self.ft.add_layer()
result = self.ft.get_leaves()
self.assertEqual(result, self.ft.get_layer(3))
def test_6(self):
for i in range(3):
self.ft.add_layer()
with self.assertRaises(ValueError):
self.ft.get_layer(4)
def test_7(self):
self.ft.add_layer()
self.ft.add_layer(lambda n: True if n.fractal_order > 1 else False)
self.ft.add_layer(lambda n: True if n.fractal_order > 1 else False)
self.ft.add_layer(lambda n: True if n.fractal_order > 1 else False)
self.ft.add_layer(lambda n: True if n.fractal_order > 1 else False)
result = [[['1.1'],
[['1.2.1'], ['1.2.2.1', '1.2.2.2', '1.2.2.3'],
['1.2.3.1', '1.2.3.2', '1.2.3.3']],
[['1.3.1.1', '1.3.1.2', '1.3.1.3'], ['1.3.2'],
['1.3.3.1', '1.3.3.2', '1.3.3.3']]], '2',
[[['3.1.1'], ['3.1.2.1', '3.1.2.2', '3.1.2.3'],
['3.1.3.1', '3.1.3.2', '3.1.3.3']],
[['3.2.1.1', '3.2.1.2', '3.2.1.3'], ['3.2.2'],
['3.2.3.1', '3.2.3.2', '3.2.3.3']], ['3.3']]]
self.assertEqual(result,
[name for name in self.ft.get_layer(4, key='name')])
def test_7_1(self):
self.ft.add_layer()
self.ft.add_layer()
self.assertEqual([10], self.ft.get_layer(0, key='value'))
def test_7_2(self):
self.ft.add_layer()
self.ft.add_layer()
result = [Fraction(5, 1), Fraction(5, 3), Fraction(10, 3)]
self.assertEqual(result, self.ft.get_layer(1, key='value'))
def test_7_3(self):
self.ft.add_layer()
self.ft.add_layer()
result = [[Fraction(5, 6),
Fraction(5, 3),
Fraction(5, 2)],
[Fraction(5, 6),
Fraction(5, 18),
Fraction(5, 9)],
[Fraction(10, 9),
Fraction(5, 3),
Fraction(5, 9)]]
self.assertEqual(result, self.ft.get_layer(2, key='value'))
def test_get_layer_key_lambda(self):
self.ft.add_layer()
self.ft.add_layer()
result = [[0.83, 1.67, 2.5], [0.83, 0.28, 0.56], [1.11, 1.67, 0.56]]
self.assertEqual(
result,
self.ft.get_layer(2, key=lambda node: round(float(node.value), 2)))