ml_pipeline_cookiecutter/{{cookiecutter.project_name}}/test/test_cnn.py

from {{cookiecutter.module_name}} import config
from {{cookiecutter.module_name}}.model.cnn import VGG11

def test_in_channels():
    assert config.model.name == 'vgg11'