How to use the foremast.utils.get_template function in foremast

def test_ec2_iam_policy():
    """Check template for proper format."""
    ec2_json = get_template(EC2_TEMPLATE_NAME)
    ec2_dict = json.loads(ec2_json)

    assert all(key in ec2_dict for key in ('Version', 'Statement'))
    assert len(ec2_dict['Statement']) == 1

    only_statement = ec2_dict['Statement'][0]

    assert only_statement['Action'] == 'sts:AssumeRole'
    assert only_statement['Effect'] == 'Allow'
    assert only_statement['Principal']['Service'] == 'ec2.amazonaws.com'

def valid_json(template, data):
    parsed_template = get_template(template_file=template, data=data)

    assert type(json.loads(parsed_template)) == dict

def test_get_template():
    with pytest.raises(ForemastTemplateNotFound):
        template = get_template(template_file='doesnotexist.json.j2')

def test_lambda_iam_policy():
    """Check Lambda Trust Relationship template format."""
    lambda_json = get_template(LAMBDA_TEMPLATE_NAME)
    lambda_dict = json.loads(lambda_json)

    assert all(key in lambda_dict for key in ('Version', 'Statement'))
    assert len(lambda_dict['Statement']) == 1

    only_statement = lambda_dict['Statement'][0]

    assert only_statement['Action'] == 'sts:AssumeRole'
    assert only_statement['Effect'] == 'Allow'
    assert only_statement['Principal']['Service'] == 'lambda.amazonaws.com'

def test_ec2_iam_policy():
    """Check template for proper format."""
    ec2_json = get_template(EC2_TEMPLATE_NAME)
    ec2_dict = json.loads(ec2_json)

    assert all(key in ec2_dict for key in ('Version', 'Statement'))
    assert len(ec2_dict['Statement']) == 1

    only_statement = ec2_dict['Statement'][0]

    assert only_statement['Action'] == 'sts:AssumeRole'
    assert only_statement['Effect'] == 'Allow'
    assert only_statement['Principal']['Service'] == 'ec2.amazonaws.com'

def test_lambda_iam_policy():
    """Check Lambda Trust Relationship template format."""
    lambda_json = get_template(LAMBDA_TEMPLATE_NAME)
    lambda_dict = json.loads(lambda_json)

    assert all(key in lambda_dict for key in ('Version', 'Statement'))
    assert len(lambda_dict['Statement']) == 1

    only_statement = lambda_dict['Statement'][0]

    assert only_statement['Action'] == 'sts:AssumeRole'
    assert only_statement['Effect'] == 'Allow'
    assert only_statement['Principal']['Service'] == 'lambda.amazonaws.com'