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

class {{ exercise | camel_case }}Test(unittest.TestCase):
    {% for case in cases -%}
    def test_{{ case["description"] | to_snake }}(self):
        {%- set seconds = case["input"]["seconds"] %}
        {%- set planet = case["input"]["planet"] %}
        self.assertEqual(SpaceAge({{ seconds }}).on_{{ planet | lower }}(), {{ case["expected"] }})

    {% endfor %}

{{ macros.footer() }}
