예제 #1
0
def test_encode_images_verbose_true(capsys):
    cnn = CNN(verbose=True)
    cnn.encode_images(image_dir=TEST_IMAGE_DIR)
    out, err = capsys.readouterr()

    assert '[==============================]' in out
    assert '' == err
예제 #2
0
def test_encode_images_verbose_false(capsys):
    cnn = CNN(verbose=False)
    cnn.encode_images(image_dir=TEST_IMAGE_DIR)
    out, err = capsys.readouterr()

    assert '' == out
    assert '' == err