f4503bb3a3be014b452f54d8e2d187bb6419f627,tests/test_utils.py,,test_model_trainer,#,7

Before Change

def test_model_trainer():
    model = RandomConvClassifierGenerator(3, (28, 28, 1)).generate()
    ModelTrainer(model, np.random.rand(2, 28, 28, 1), np.random.rand(2, 3), np.random.rand(1, 28, 28, 1),
                 np.random.rand(1, 3), False).train_model()


def test_model_trainer_not_augmented():
    constant.DATA_AUGMENTATION = False

After Change

def test_model_trainer():
    model = DefaultClassifierGenerator(3, (28, 28, 3)).generate().produce_model()
    train_data, test_data = get_processed_data()
    ModelTrainer(model, train_data, test_data, False).train_model(max_iter_num=3)

Instances