| /* |
| * Copyright (C) 2018 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #ifndef LIBTEXTCLASSIFIER_ANNOTATOR_NUMBER_NUMBER_TEST_INCLUDE_H_ |
| #define LIBTEXTCLASSIFIER_ANNOTATOR_NUMBER_NUMBER_TEST_INCLUDE_H_ |
| |
| #include "annotator/model_generated.h" |
| #include "annotator/number/number.h" |
| #include "utils/jvm-test-utils.h" |
| #include "gtest/gtest.h" |
| |
| namespace libtextclassifier3 { |
| namespace test_internal { |
| |
| class NumberAnnotatorTest : public ::testing::Test { |
| private: |
| protected: |
| explicit NumberAnnotatorTest(ModeFlag enabled_modes = ModeFlag_ALL) |
| : unilib_(CreateUniLibForTesting()), |
| number_annotator_(TestingNumberAnnotatorOptions(enabled_modes), |
| unilib_.get()) {} |
| |
| const NumberAnnotatorOptions* TestingNumberAnnotatorOptions( |
| ModeFlag enabled_modes); |
| |
| std::unique_ptr<UniLib> unilib_; |
| NumberAnnotator number_annotator_; |
| }; |
| |
| class NumberAnnotatorForAnnotationAndClassificationTest |
| : public NumberAnnotatorTest { |
| protected: |
| NumberAnnotatorForAnnotationAndClassificationTest() |
| : NumberAnnotatorTest(ModeFlag_ANNOTATION_AND_CLASSIFICATION) {} |
| }; |
| |
| class NumberAnnotatorForSelectionTest : public NumberAnnotatorTest { |
| protected: |
| NumberAnnotatorForSelectionTest() : NumberAnnotatorTest(ModeFlag_SELECTION) {} |
| }; |
| |
| } // namespace test_internal |
| } // namespace libtextclassifier3 |
| |
| #endif // LIBTEXTCLASSIFIER_ANNOTATOR_NUMBER_NUMBER_TEST_INCLUDE_H_ |