Update TestCollections.java

master
agibsonccc 2021-03-26 00:22:38 +09:00
parent 2213d445b1
commit 78e32abebc
1 changed files with 7 additions and 3 deletions

View File

@ -20,23 +20,27 @@
package org.nd4j.linalg.util; 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.ParameterizedTest;
import org.junit.jupiter.params.provider.MethodSource; 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.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 org.nd4j.linalg.factory.Nd4jBackend;
import java.util.*; import java.util.*;
import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue; import static org.junit.jupiter.api.Assertions.assertTrue;
@NativeTag @NativeTag
public class TestCollections extends BaseNd4jTestWithBackends { public class TestCollections extends BaseNd4jTestWithBackends {
@ParameterizedTest @ParameterizedTest
@MethodSource("org.nd4j.linalg.BaseNd4jTestWithBackends#configs") @MethodSource("org.nd4j.linalg.BaseNd4jTestWithBackends#configs")
@Tag(TagNames.LONG_TEST)
@Tag(TagNames.LARGE_RESOURCES)
public void testCompactHeapStringList(Nd4jBackend backend) { public void testCompactHeapStringList(Nd4jBackend backend) {
int[] reallocSizeBytes = new int[] {1024, 1048576}; int[] reallocSizeBytes = new int[] {1024, 1048576};