From 0e4be5c4d26afb68092af9d0c0f9a0f1a4f010d1 Mon Sep 17 00:00:00 2001 From: brian Date: Wed, 9 Aug 2023 12:13:31 +0200 Subject: [PATCH] Tag some junit tests with @Tag("long-running") to skip them during normal build Signed-off-by: brian --- brutex-extended-tests/src/test/java/net/brutex/gan/App.java | 3 ++- brutex-extended-tests/src/test/java/net/brutex/gan/App2.java | 3 ++- .../src/test/java/net/brutex/gan/MnistSimpleGAN.java | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/brutex-extended-tests/src/test/java/net/brutex/gan/App.java b/brutex-extended-tests/src/test/java/net/brutex/gan/App.java index dc5898413..6395223ad 100644 --- a/brutex-extended-tests/src/test/java/net/brutex/gan/App.java +++ b/brutex-extended-tests/src/test/java/net/brutex/gan/App.java @@ -16,6 +16,7 @@ import org.deeplearning4j.nn.conf.layers.misc.FrozenLayerWithBackprop; import org.deeplearning4j.nn.multilayer.MultiLayerNetwork; import org.deeplearning4j.nn.weights.WeightInit; import org.deeplearning4j.optimize.listeners.PerformanceListener; +import org.junit.jupiter.api.Tag; import org.junit.jupiter.api.Test; import org.nd4j.linalg.activations.Activation; import org.nd4j.linalg.activations.impl.ActivationLReLU; @@ -122,7 +123,7 @@ public class App { return conf; } - @Test + @Test @Tag("long-running") public void runTest() throws Exception { App.main(null); } diff --git a/brutex-extended-tests/src/test/java/net/brutex/gan/App2.java b/brutex-extended-tests/src/test/java/net/brutex/gan/App2.java index f9f7aba58..73e271fe1 100644 --- a/brutex-extended-tests/src/test/java/net/brutex/gan/App2.java +++ b/brutex-extended-tests/src/test/java/net/brutex/gan/App2.java @@ -44,6 +44,7 @@ import org.deeplearning4j.nn.conf.layers.*; import org.deeplearning4j.nn.conf.layers.misc.FrozenLayerWithBackprop; import org.deeplearning4j.nn.multilayer.MultiLayerNetwork; import org.deeplearning4j.optimize.listeners.PerformanceListener; +import org.junit.jupiter.api.Tag; import org.junit.jupiter.api.Test; import org.nd4j.linalg.api.ndarray.INDArray; import org.nd4j.linalg.dataset.DataSet; @@ -70,7 +71,7 @@ public class App2 { final static INDArray label_real = Nd4j.ones(BATCHSIZE, 1); final static INDArray label_fake = Nd4j.zeros(BATCHSIZE, 1); - @Test + @Test @Tag("long-running") void runTest() throws IOException { Nd4j.getMemoryManager().setAutoGcWindow(15 * 1000); diff --git a/brutex-extended-tests/src/test/java/net/brutex/gan/MnistSimpleGAN.java b/brutex-extended-tests/src/test/java/net/brutex/gan/MnistSimpleGAN.java index 2cfddb795..a31e9298f 100644 --- a/brutex-extended-tests/src/test/java/net/brutex/gan/MnistSimpleGAN.java +++ b/brutex-extended-tests/src/test/java/net/brutex/gan/MnistSimpleGAN.java @@ -31,6 +31,7 @@ import org.deeplearning4j.nn.conf.layers.DropoutLayer; import org.deeplearning4j.nn.conf.layers.OutputLayer; import org.deeplearning4j.nn.multilayer.MultiLayerNetwork; import org.deeplearning4j.nn.weights.WeightInit; +import org.junit.jupiter.api.Tag; import org.junit.jupiter.api.Test; import org.nd4j.linalg.activations.Activation; import org.nd4j.linalg.activations.impl.ActivationLReLU; @@ -100,7 +101,7 @@ public class MnistSimpleGAN { return new MultiLayerNetwork(discConf); } - @Test + @Test @Tag("long-running") public void runTest() throws Exception { main(null); }