
import pytest

from cookiecutter.environment import StrictEnvironment
from cookiecutter.exceptions import UnknownExtension


def test_env_should_raise_for_unknown_extension():
    context = {
        'cookiecutter': {
            '_extensions': ['foobar']
        }
    }

    with pytest.raises(UnknownExtension) as err:
        StrictEnvironment(context=context, keep_trailing_newline=True)

    assert 'Unable to load extension: ' in str(err.value)


import pytest

from cookiecutter.environment import StrictEnvironment
from cookiecutter.exceptions import UnknownExtension


def test_env_should_raise_for_unknown_extension():
    context = {
        'cookiecutter': {
            '_extensions': ['foobar']
        }
    }

    with pytest.raises(UnknownExtension) as err:
        StrictEnvironment(context=context, keep_trailing_newline=True)

    assert 'Unable to load extension: ' in str(err.value)


def test_env_should_come_with_jinja2_time_extension():
    env = StrictEnvironment(keep_trailing_newline=True)
    assert 'jinja2_time.jinja2_time.TimeExtension' in env.extensions

Add a simple test to make sure cookiecutter comes with jinja2_time
Add a simple test to make sure cookiecutter comes with jinja2_time

  
  import pytest
  
  from cookiecutter.environment import StrictEnvironment
  from cookiecutter.exceptions import UnknownExtension
  
  
  def test_env_should_raise_for_unknown_extension():
      context = {
          'cookiecutter': {
              '_extensions': ['foobar']
          }
      }
  
      with pytest.raises(UnknownExtension) as err:
          StrictEnvironment(context=context, keep_trailing_newline=True)
  
      assert 'Unable to load extension: ' in str(err.value)
  
+ 
+ def test_env_should_come_with_jinja2_time_extension():
+     env = StrictEnvironment(keep_trailing_newline=True)
+     assert 'jinja2_time.jinja2_time.TimeExtension' in env.extensions
+ 
Add a simple test to make sure cookiecutter comes with jinja2_time
## Code Before:

import pytest

from cookiecutter.environment import StrictEnvironment
from cookiecutter.exceptions import UnknownExtension


def test_env_should_raise_for_unknown_extension():
    context = {
        'cookiecutter': {
            '_extensions': ['foobar']
        }
    }

    with pytest.raises(UnknownExtension) as err:
        StrictEnvironment(context=context, keep_trailing_newline=True)

    assert 'Unable to load extension: ' in str(err.value)

## Instruction:
Add a simple test to make sure cookiecutter comes with jinja2_time
## Code After:

import pytest

from cookiecutter.environment import StrictEnvironment
from cookiecutter.exceptions import UnknownExtension


def test_env_should_raise_for_unknown_extension():
    context = {
        'cookiecutter': {
            '_extensions': ['foobar']
        }
    }

    with pytest.raises(UnknownExtension) as err:
        StrictEnvironment(context=context, keep_trailing_newline=True)

    assert 'Unable to load extension: ' in str(err.value)


def test_env_should_come_with_jinja2_time_extension():
    env = StrictEnvironment(keep_trailing_newline=True)
    assert 'jinja2_time.jinja2_time.TimeExtension' in env.extensions
