def METHOD_NAME(self):
    text_val = np.array([["a", "Test 1 2 3"], ["Hi there", "test test"]], dtype=np.str)
    def func(text):
        x = tf.strings.split(text, sep=' ').flat_values
        x_ = tf.identity(x, name=_TFOUTPUT)
        return x_
    self._run_test_case(func, [_OUTPUT], {_INPUT: text_val})