|
| 1 | +{%- import "generator_macros.j2" as macros with context -%} |
| 2 | +{{ macros.canonical_ref() }} |
| 3 | + |
| 4 | +{{ macros.header(imports=['RelativeDistance']) }} |
| 5 | + |
| 6 | +class {{ exercise | camel_case }}Test(unittest.TestCase): |
| 7 | + {% for case in cases -%} |
| 8 | + def test_{{ case["description"] | to_snake }}(self): |
| 9 | + family_tree = { |
| 10 | + {%- for person, relatives in case["input"]["familyTree"].items() %} |
| 11 | + "{{ person }}": {{ relatives }}, |
| 12 | + {%- endfor %} |
| 13 | + } |
| 14 | + {%- if case["expected"] is none %} |
| 15 | + with self.assertRaises(ValueError): |
| 16 | + RelativeDistance(family_tree).degree_of_separation("{{ case["input"]["personA"] }}", "{{ case["input"]["personB"] }}") |
| 17 | + {%- else %} |
| 18 | + self.assertEqual( |
| 19 | + RelativeDistance(family_tree).degree_of_separation("{{ case["input"]["personA"] }}", "{{ case["input"]["personB"] }}"), |
| 20 | + {{ case["expected"] }} |
| 21 | + ) |
| 22 | + {%- endif %} |
| 23 | + {% endfor -%} |
| 24 | + |
| 25 | + # Additional track-specific tests |
| 26 | + {% for case in additional_cases -%} |
| 27 | + def test_{{ case["description"] | to_snake }}(self): |
| 28 | + family_tree = { |
| 29 | + {%- for person, relatives in case["input"]["family_tree"].items() %} |
| 30 | + "{{ person }}": {{ relatives }}, |
| 31 | + {%- endfor %} |
| 32 | + } |
| 33 | + {%- if case["expected"]["error"] %} |
| 34 | + with self.assertRaises(ValueError) as err: |
| 35 | + RelativeDistance(family_tree).degree_of_separation("{{ case["input"]["person_a"] }}", "{{ case["input"]["person_b"] }}") |
| 36 | + self.assertEqual(type(err.exception), ValueError) |
| 37 | + self.assertEqual(err.exception.args[0], "{{ case["expected"]["error"] }}") |
| 38 | + {%- else %} |
| 39 | + self.assertEqual( |
| 40 | + RelativeDistance(family_tree).degree_of_separation("{{ case["input"]["person_a"] }}", "{{ case["input"]["person_b"] }}"), |
| 41 | + {{ case["expected"] }} |
| 42 | + ) |
| 43 | + {%- endif %} |
| 44 | + {% endfor -%} |
0 commit comments