diff options
Diffstat (limited to 'tensorflow/java/src/test/java/org/tensorflow/processor/OperatorProcessorTest.java')
-rw-r--r-- | tensorflow/java/src/test/java/org/tensorflow/processor/OperatorProcessorTest.java | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/tensorflow/java/src/test/java/org/tensorflow/processor/OperatorProcessorTest.java b/tensorflow/java/src/test/java/org/tensorflow/processor/OperatorProcessorTest.java new file mode 100644 index 0000000000..9fa1bad20d --- /dev/null +++ b/tensorflow/java/src/test/java/org/tensorflow/processor/OperatorProcessorTest.java @@ -0,0 +1,51 @@ +/* Copyright 2017 The TensorFlow Authors. All Rights Reserved. + +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. +==============================================================================*/ + +package org.tensorflow.processor; + +import static com.google.testing.compile.CompilationSubject.assertThat; + +import com.google.testing.compile.Compilation; +import com.google.testing.compile.Compiler; +import com.google.testing.compile.JavaFileObjects; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.junit.runners.JUnit4; + +/** Basic tests for {@link org.tensorflow.processor.operator.OperatorProcessor}. */ +@RunWith(JUnit4.class) +public final class OperatorProcessorTest { + + @Test + public void basicGood() { + Compilation compile = compile("org/tensorflow/processor/operator/good/BasicGood.java"); + assertThat(compile).succeededWithoutWarnings(); + assertThat(compile).generatedSourceFile("org.tensorflow.op.Ops"); + } + + @Test + public void basicBad() { + assertThat(compile("org/tensorflow/processor/operator/bad/BasicBad.java")).failed(); + } + + // Create a compilation unit that includes the @Operator annotation and processor. + private static Compilation compile(String path) { + return Compiler.javac() + .withProcessors(new OperatorProcessor()) + .compile( + JavaFileObjects.forResource("src/main/java/org/tensorflow/op/annotation/Operator.java"), + JavaFileObjects.forResource(path)); + } +} |