Update NormalizerSerializerTest.java

master
agibsonccc 2021-03-26 09:27:31 +09:00
parent b874696c64
commit b3805a132f
1 changed files with 4 additions and 4 deletions

View File

@ -21,11 +21,10 @@
package org.nd4j.linalg.dataset; package org.nd4j.linalg.dataset;
import lombok.Getter; import lombok.Getter;
import org.junit.jupiter.api.BeforeAll; import org.junit.jupiter.api.*;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Tag;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.io.TempDir; import org.junit.jupiter.api.io.TempDir;
import org.junit.jupiter.api.parallel.Execution;
import org.junit.jupiter.api.parallel.ExecutionMode;
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;
@ -296,6 +295,7 @@ public class NormalizerSerializerTest extends BaseNd4jTestWithBackends {
@ParameterizedTest @ParameterizedTest
@MethodSource("org.nd4j.linalg.BaseNd4jTestWithBackends#configs") @MethodSource("org.nd4j.linalg.BaseNd4jTestWithBackends#configs")
@Execution(ExecutionMode.SAME_THREAD)
public void testCustomNormalizerWithoutRegisteredStrategy(Nd4jBackend backend) throws Exception { public void testCustomNormalizerWithoutRegisteredStrategy(Nd4jBackend backend) throws Exception {
assertThrows(RuntimeException.class, () -> { assertThrows(RuntimeException.class, () -> {
File normalizerFile = Files.createTempFile(tmpFile.toPath(),"pre-process-" + UUID.randomUUID().toString(),"bin").toFile(); File normalizerFile = Files.createTempFile(tmpFile.toPath(),"pre-process-" + UUID.randomUUID().toString(),"bin").toFile();