def test_set_untrainable_on_dense3():
model = create_dense_model((33, ), 27, 41)
set_untrainable(model, 1)
assert get_trainable_layers(model) == 1148
def test_set_untrainable_on_dense2():
model = create_dense_model((10, ), 10, 10)
set_untrainable(model, 0)
assert get_trainable_layers(model) == 0
def test_set_untrainable_on_cnn4():
model = create_cnn_model((28, 28, 1), 10, 10)
set_untrainable(model, 0)
assert get_trainable_layers(model) == 0
def test_set_untrainable_with_negative_argument():
model = create_cnn_model((28, 28, 1), 10, 10)
set_untrainable(model, -1)
assert get_trainable_layers(model) == 5761
def test_set_untrainable_on_cnn3():
model = create_cnn_model((28, 28, 1), 10, 10)
set_untrainable(model, 4)
assert get_trainable_layers(model) == 5761 + 910 + 100
