1 /* 2 * Copyright (C) 2017 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #ifndef LIBTEXTCLASSIFIER_TYPES_TEST_UTIL_H_ 18 #define LIBTEXTCLASSIFIER_TYPES_TEST_UTIL_H_ 19 20 #include <ostream> 21 22 #include "types.h" 23 #include "util/base/logging.h" 24 25 namespace libtextclassifier2 { 26 27 inline std::ostream& operator<<(std::ostream& stream, const Token& value) { 28 logging::LoggingStringStream tmp_stream; 29 tmp_stream << value; 30 return stream << tmp_stream.message; 31 } 32 33 inline std::ostream& operator<<(std::ostream& stream, 34 const AnnotatedSpan& value) { 35 logging::LoggingStringStream tmp_stream; 36 tmp_stream << value; 37 return stream << tmp_stream.message; 38 } 39 40 inline std::ostream& operator<<(std::ostream& stream, 41 const DatetimeParseResultSpan& value) { 42 logging::LoggingStringStream tmp_stream; 43 tmp_stream << value; 44 return stream << tmp_stream.message; 45 } 46 47 } // namespace libtextclassifier2 48 49 #endif // LIBTEXTCLASSIFIER_TYPES_TEST_UTIL_H_ 50