diff --git a/tensorflow/python/keras/layers/preprocessing/text_vectorization_test.py b/tensorflow/python/keras/layers/preprocessing/text_vectorization_test.py index 1a850b171af..0202cebde73 100644 --- a/tensorflow/python/keras/layers/preprocessing/text_vectorization_test.py +++ b/tensorflow/python/keras/layers/preprocessing/text_vectorization_test.py @@ -469,25 +469,25 @@ class TextVectorizationPreprocessingTest( layer._split = "unsuppported" with self.assertRaisesRegex(ValueError, ".*is not a supported splitting.*"): _ = layer(input_data) - + def test_standardize_with_no_identical_argument(self): input_array = np.array([["hello world"]]) expected_output = np.array([[1, 1]]) - + standardize = "".join(["lower", "_and_strip_punctuation"]) layer = get_layer_class()(standardize=standardize) output = layer(input_array).numpy() - + self.assertAllEqual(expected_output, output) - + def test_splitting_with_no_identical_argument(self): input_array = np.array([["hello world"]]) expected_output = np.array([[1, 1]]) - + split = "".join(["white", "space"]) layer = get_layer_class()(split=split) output = layer(input_array).numpy() - + self.assertAllEqual(expected_output, output) @keras_parameterized.run_all_keras_modes