diff --git a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/util/TestCollections.java b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/util/TestCollections.java index 4f396dcf8..ee9755353 100644 --- a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/util/TestCollections.java +++ b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/util/TestCollections.java @@ -20,23 +20,27 @@ package org.nd4j.linalg.util; -import org.junit.jupiter.api.Test; +import org.junit.jupiter.api.Tag; import org.junit.jupiter.params.ParameterizedTest; import org.junit.jupiter.params.provider.MethodSource; -import org.nd4j.common.tests.tags.NativeTag; -import org.nd4j.linalg.BaseNd4jTestWithBackends; import org.nd4j.common.collection.CompactHeapStringList; +import org.nd4j.common.tests.tags.NativeTag; +import org.nd4j.common.tests.tags.TagNames; +import org.nd4j.linalg.BaseNd4jTestWithBackends; import org.nd4j.linalg.factory.Nd4jBackend; import java.util.*; import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertTrue; + @NativeTag public class TestCollections extends BaseNd4jTestWithBackends { @ParameterizedTest @MethodSource("org.nd4j.linalg.BaseNd4jTestWithBackends#configs") + @Tag(TagNames.LONG_TEST) + @Tag(TagNames.LARGE_RESOURCES) public void testCompactHeapStringList(Nd4jBackend backend) { int[] reallocSizeBytes = new int[] {1024, 1048576};