generator_macros.j2

{# Usage: {%- import "generator_macros.j2" as macros -%} #}
{# {{ macros.linebreak(text) }} #}

{%- macro linebreak(s) %}
{%- set parts = s.split(": ") -%}
"{{ parts[0] }}: "
{% for part in parts[1].split(", ") %}
"{{ part }}{% if not loop.last %}, {% endif %}"
{%- endfor %}
{% endmacro -%}

{% macro canonical_ref() -%}
# These tests are auto-generated with test data from:
# https://github.com/exercism/problem-specifications/tree/main/exercises/{{ exercise }}/canonical-data.json
# File last updated on {{ current_date }}
{%- endmacro %}

{% macro header(imports=[], ignore=[]) -%}

import unittest

from {{ exercise | to_snake }} import ({% if imports -%}
{% for name in imports -%}
    {{ name }},
{% endfor %}
{%- else -%}
{% for prop in properties -%}
    {%- if prop not in ignore -%}
    {{ prop | to_snake }},
    {%- endif -%}
{% endfor %}
{%- endif %})
{%- endmacro %}

{% macro utility() -%}# Utility functions
    def assertRaisesWithMessage(self, exception):
        return self.assertRaisesRegex(exception, r".+")
{%- endmacro %}

{% macro empty_set(set, list, class_name) -%}
{%- if list|length > 0 -%}
{{ set }} = {{ class_name }}({{ list }})
{%- else -%}
{{ set }} = {{ class_name }}()
{%- endif %}
{%- endmacro %}