Skip to content
Snippets Groups Projects
Commit 2f33d2a9 authored by Falco Weichselbaum's avatar Falco Weichselbaum
Browse files

Changed assertion comparisment containing old keras declarations. Disabled...

Changed assertion comparisment containing old keras declarations. Disabled testing '_compare_keras_optimizers' since it is disabled
parent 50314156
No related branches found
No related tags found
4 merge requests!413update release branch,!412Resolve "release v2.0.0",!349trying a new requirements.txt list that was generated using pipreqs (an...,!335Resolve "upgrade code to TensorFlow V2"
Pipeline #82011 failed
import tensorflow.keras as keras
import keras
import pytest
from mlair import AbstractModelClass
......@@ -145,33 +145,35 @@ class TestAbstractModelClass:
with pytest.raises(ValueError) as einfo:
amc.compile_options = {"optimizer": keras.optimizers.Adam()}
assert "Got different values or arguments for same argument: self.optimizer=<class" \
" 'keras.optimizers.SGD'> and 'optimizer': <class 'keras.optimizers.Adam'>" in str(einfo.value)
" 'tensorflow.python.keras.optimizer_v2.gradient_descent.SGD'> and " \
"'optimizer': <class 'tensorflow.python.keras.optimizer_v2.adam.Adam'>" in str(einfo.value)
def test_compile_options_setter_as_mix_attr_dict_invalid_duplicates_same_optimizer_other_args(self, amc):
amc.optimizer = keras.optimizers.SGD(lr=0.1)
with pytest.raises(ValueError) as einfo:
amc.compile_options = {"optimizer": keras.optimizers.SGD(lr=0.001)}
assert "Got different values or arguments for same argument: self.optimizer=<class" \
" 'keras.optimizers.SGD'> and 'optimizer': <class 'keras.optimizers.SGD'>" in str(einfo.value)
" 'tensorflow.python.keras.optimizer_v2.gradient_descent.SGD'> and " \
"'optimizer': <class 'tensorflow.python.keras.optimizer_v2.gradient_descent.SGD'>" in str(einfo.value)
def test_compile_options_setter_as_dict_invalid_keys(self, amc):
with pytest.raises(ValueError) as einfo:
amc.compile_options = {"optimizer": keras.optimizers.SGD(), "InvalidKeyword": [1, 2, 3]}
assert "Got invalid key for compile_options. dict_keys(['optimizer', 'InvalidKeyword'])" in str(einfo.value)
def test_compare_keras_optimizers_equal(self, amc):
assert amc._AbstractModelClass__compare_keras_optimizers(keras.optimizers.SGD(), keras.optimizers.SGD()) is True
def test_compare_keras_optimizers_no_optimizer(self, amc):
assert amc._AbstractModelClass__compare_keras_optimizers('NoOptimizer', keras.optimizers.SGD()) is False
def test_compare_keras_optimizers_other_parameters_run_sess(self, amc):
assert amc._AbstractModelClass__compare_keras_optimizers(keras.optimizers.SGD(lr=0.1),
keras.optimizers.SGD(lr=0.01)) is False
def test_compare_keras_optimizers_other_parameters_none_sess(self, amc):
assert amc._AbstractModelClass__compare_keras_optimizers(keras.optimizers.SGD(decay=1),
keras.optimizers.SGD(decay=0.01)) is False
# def test_compare_keras_optimizers_equal(self, amc):
# assert amc._AbstractModelClass__compare_keras_optimizers(keras.optimizers.SGD(), keras.optimizers.SGD()) is True
#
# def test_compare_keras_optimizers_no_optimizer(self, amc):
# assert amc._AbstractModelClass__compare_keras_optimizers('NoOptimizer', keras.optimizers.SGD()) is False
#
# def test_compare_keras_optimizers_other_parameters_run_sess(self, amc):
# assert amc._AbstractModelClass__compare_keras_optimizers(keras.optimizers.SGD(lr=0.1),
# keras.optimizers.SGD(lr=0.01)) is False
#
# def test_compare_keras_optimizers_other_parameters_none_sess(self, amc):
# assert amc._AbstractModelClass__compare_keras_optimizers(keras.optimizers.SGD(decay=1),
# keras.optimizers.SGD(decay=0.01)) is False
def test_getattr(self, amc):
amc.model = keras.Model()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment