ml_pipeline/test/test_cnn.py

from ml_pipeline import config
from ml_pipeline.model.cnn import VGG11

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