{%- import "generator_macros.j2" as macros with context -%}
{{ macros.header() }}

class {{ exercise | camel_case }}Test(unittest.TestCase):
    {% for supercase in cases %}
        {% for case in supercase["cases"] -%}
        def test_{{ case["description"] | to_snake }}(self):
            {%- set number = case["input"]["number"] %}
            {%- set expected = case["expected"] %}
            self.assertEqual({{ case["property"] | to_snake }}({{ number }}), {{ expected }})
            
        {% endfor %}
    {% endfor %}

{{ macros.footer() }}