added test for normalize_data_format in keras_utils (#1818)

* added test for normalize_data_format in keras_utils * change after review
tensorflow · May 13, 2020 · c05a3d6 · c05a3d6
1 parent 88ec116
commit c05a3d6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tensorflow_addons/utils/tests/keras_utils_test.py b/tensorflow_addons/utils/tests/keras_utils_test.py
@@ -22,6 +22,14 @@
 from tensorflow_addons.utils import keras_utils
 
 
+def test_normalize_data_format():
+    assert keras_utils.normalize_data_format("Channels_Last") == "channels_last"
+    assert keras_utils.normalize_data_format("CHANNELS_FIRST") == "channels_first"
+
+    with pytest.raises(ValueError, match="The `data_format` argument must be one of"):
+        keras_utils.normalize_data_format("invalid")
+
+
 def test_normalize_tuple():
     assert (2, 2, 2) == keras_utils.normalize_tuple(2, n=3, name="strides")
     assert (2, 1, 2) == keras_utils.normalize_tuple((2, 1, 2), n=3, name="strides")