be794198e1
* Corrected small typos around resistor bands. * Due to failing CI, alterations to the test generator script were needed. The generated vs submitted diff now skips the first three lines of the file so that the generation date is not picked up and flagged as needing regeneration. Sadly, a workaround was also needed to prevent Python difflib from noting the difference anyways and producing an empty "false positive" diff. All templates and test files also needed to be altered to ensure that the first three lines of every test file will always be the autogeneration comment and date. Hopefully, this will now stop the CI failures without creating any subtle additional bugs. * Touch up to bowling template. Added back the error raising utility. * Touch up to two-bucket template to add back in error raising utility. [no important files changed]
174 lines
6.2 KiB
Python
174 lines
6.2 KiB
Python
# These tests are auto-generated with test data from:
|
|
# https://github.com/exercism/problem-specifications/tree/main/exercises/pov/canonical-data.json
|
|
# File last updated on 2023-07-19
|
|
|
|
import unittest
|
|
|
|
from pov import (
|
|
Tree,
|
|
)
|
|
|
|
|
|
class PovTest(unittest.TestCase):
|
|
def test_results_in_the_same_tree_if_the_input_tree_is_a_singleton(self):
|
|
tree = Tree("x")
|
|
expected = Tree("x")
|
|
self.assertTreeEquals(tree.from_pov("x"), expected)
|
|
|
|
def test_can_reroot_a_tree_with_a_parent_and_one_sibling(self):
|
|
tree = Tree("parent", [Tree("x"), Tree("sibling")])
|
|
expected = Tree("x", [Tree("parent", [Tree("sibling")])])
|
|
self.assertTreeEquals(tree.from_pov("x"), expected)
|
|
|
|
def test_can_reroot_a_tree_with_a_parent_and_many_siblings(self):
|
|
tree = Tree("parent", [Tree("a"), Tree("x"), Tree("b"), Tree("c")])
|
|
expected = Tree("x", [Tree("parent", [Tree("a"), Tree("b"), Tree("c")])])
|
|
self.assertTreeEquals(tree.from_pov("x"), expected)
|
|
|
|
def test_can_reroot_a_tree_with_new_root_deeply_nested_in_tree(self):
|
|
tree = Tree(
|
|
"level-0",
|
|
[Tree("level-1", [Tree("level-2", [Tree("level-3", [Tree("x")])])])],
|
|
)
|
|
expected = Tree(
|
|
"x",
|
|
[Tree("level-3", [Tree("level-2", [Tree("level-1", [Tree("level-0")])])])],
|
|
)
|
|
self.assertTreeEquals(tree.from_pov("x"), expected)
|
|
|
|
def test_moves_children_of_the_new_root_to_same_level_as_former_parent(self):
|
|
tree = Tree("parent", [Tree("x", [Tree("kid-0"), Tree("kid-1")])])
|
|
expected = Tree("x", [Tree("kid-0"), Tree("kid-1"), Tree("parent")])
|
|
self.assertTreeEquals(tree.from_pov("x"), expected)
|
|
|
|
def test_can_reroot_a_complex_tree_with_cousins(self):
|
|
tree = Tree(
|
|
"grandparent",
|
|
[
|
|
Tree(
|
|
"parent",
|
|
[
|
|
Tree("x", [Tree("kid-0"), Tree("kid-1")]),
|
|
Tree("sibling-0"),
|
|
Tree("sibling-1"),
|
|
],
|
|
),
|
|
Tree("uncle", [Tree("cousin-0"), Tree("cousin-1")]),
|
|
],
|
|
)
|
|
expected = Tree(
|
|
"x",
|
|
[
|
|
Tree("kid-1"),
|
|
Tree("kid-0"),
|
|
Tree(
|
|
"parent",
|
|
[
|
|
Tree("sibling-0"),
|
|
Tree("sibling-1"),
|
|
Tree(
|
|
"grandparent",
|
|
[Tree("uncle", [Tree("cousin-0"), Tree("cousin-1")])],
|
|
),
|
|
],
|
|
),
|
|
],
|
|
)
|
|
self.assertTreeEquals(tree.from_pov("x"), expected)
|
|
|
|
def test_errors_if_target_does_not_exist_in_a_singleton_tree(self):
|
|
tree = Tree("x")
|
|
with self.assertRaises(ValueError) as err:
|
|
tree.from_pov("nonexistent")
|
|
self.assertEqual(type(err.exception), ValueError)
|
|
self.assertEqual(err.exception.args[0], "Tree could not be reoriented")
|
|
|
|
def test_errors_if_target_does_not_exist_in_a_large_tree(self):
|
|
tree = Tree(
|
|
"parent",
|
|
[
|
|
Tree("x", [Tree("kid-0"), Tree("kid-1")]),
|
|
Tree("sibling-0"),
|
|
Tree("sibling-1"),
|
|
],
|
|
)
|
|
with self.assertRaises(ValueError) as err:
|
|
tree.from_pov("nonexistent")
|
|
self.assertEqual(type(err.exception), ValueError)
|
|
self.assertEqual(err.exception.args[0], "Tree could not be reoriented")
|
|
|
|
def test_can_find_path_to_parent(self):
|
|
tree = Tree("parent", [Tree("x"), Tree("sibling")])
|
|
expected = ["x", "parent"]
|
|
self.assertEqual(tree.path_to("x", "parent"), expected)
|
|
|
|
def test_can_find_path_to_sibling(self):
|
|
tree = Tree("parent", [Tree("a"), Tree("x"), Tree("b"), Tree("c")])
|
|
expected = ["x", "parent", "b"]
|
|
self.assertEqual(tree.path_to("x", "b"), expected)
|
|
|
|
def test_can_find_path_to_cousin(self):
|
|
tree = Tree(
|
|
"grandparent",
|
|
[
|
|
Tree(
|
|
"parent",
|
|
[
|
|
Tree("x", [Tree("kid-0"), Tree("kid-1")]),
|
|
Tree("sibling-0"),
|
|
Tree("sibling-1"),
|
|
],
|
|
),
|
|
Tree("uncle", [Tree("cousin-0"), Tree("cousin-1")]),
|
|
],
|
|
)
|
|
expected = ["x", "parent", "grandparent", "uncle", "cousin-1"]
|
|
self.assertEqual(tree.path_to("x", "cousin-1"), expected)
|
|
|
|
def test_can_find_path_not_involving_root(self):
|
|
tree = Tree(
|
|
"grandparent",
|
|
[Tree("parent", [Tree("x"), Tree("sibling-0"), Tree("sibling-1")])],
|
|
)
|
|
expected = ["x", "parent", "sibling-1"]
|
|
self.assertEqual(tree.path_to("x", "sibling-1"), expected)
|
|
|
|
def test_can_find_path_from_nodes_other_than_x(self):
|
|
tree = Tree("parent", [Tree("a"), Tree("x"), Tree("b"), Tree("c")])
|
|
expected = ["a", "parent", "c"]
|
|
self.assertEqual(tree.path_to("a", "c"), expected)
|
|
|
|
def test_errors_if_destination_does_not_exist(self):
|
|
tree = Tree(
|
|
"parent",
|
|
[
|
|
Tree("x", [Tree("kid-0"), Tree("kid-1")]),
|
|
Tree("sibling-0"),
|
|
Tree("sibling-1"),
|
|
],
|
|
)
|
|
with self.assertRaises(ValueError) as err:
|
|
tree.path_to("x", "nonexistent")
|
|
self.assertEqual(type(err.exception), ValueError)
|
|
|
|
self.assertEqual(err.exception.args[0], "No path found")
|
|
|
|
def test_errors_if_source_does_not_exist(self):
|
|
tree = Tree(
|
|
"parent",
|
|
[
|
|
Tree("x", [Tree("kid-0"), Tree("kid-1")]),
|
|
Tree("sibling-0"),
|
|
Tree("sibling-1"),
|
|
],
|
|
)
|
|
with self.assertRaises(ValueError) as err:
|
|
tree.path_to("nonexistent", "x")
|
|
self.assertEqual(type(err.exception), ValueError)
|
|
|
|
self.assertEqual(err.exception.args[0], "Tree could not be reoriented")
|
|
|
|
# Custom Utility Functions
|
|
def assertTreeEquals(self, result, expected):
|
|
self.assertEqual(result, expected, "{} != {}".format(result, expected))
|