Package org.deeplearning4j

Class BaseDL4JTest

java.lang.Object

org.deeplearning4j.BaseDL4JTest

@DisplayName("Base DL 4 J Test")
public abstract class BaseDL4JTest
extends Object

Field Summary

Fields

Modifier and Type	Field	Description
protected static Boolean	integrationTest
protected long	startTime
protected int	threadCountBefore

Constructor Summary

Constructors

Constructor	Description
BaseDL4JTest()

Method Summary

Modifier and Type	Method	Description
DataType	getDataType()	Override this to set the datatype of the tests defined in the child class
DataType	getDefaultFPDataType()
OpExecutioner.ProfilingMode	getProfilingMode()	Override this to set the profiling mode for the tests defined in the child class
long	getTimeoutMilliseconds()	Override this method to set the default timeout for methods in the test class
static boolean	isIntegrationTests()
int	numThreads()	Override this to specify the number of threads for C++ execution, via Environment.setMaxMasterThreads(int)
static void	skipUnlessIntegrationTests()	Call this as the first line of a test in order to skip that test, only when the integration tests maven profile is not enabled.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

startTime

protected long startTime

threadCountBefore

protected int threadCountBefore

integrationTest

protected static Boolean integrationTest

Constructor Detail

BaseDL4JTest

public BaseDL4JTest()

Method Detail

numThreads

public int numThreads()

Override this to specify the number of threads for C++ execution, via Environment.setMaxMasterThreads(int)

Returns:

Number of threads to use for C++ op execution

getTimeoutMilliseconds

public long getTimeoutMilliseconds()

Override this method to set the default timeout for methods in the test class

getProfilingMode

public OpExecutioner.ProfilingMode getProfilingMode()

Override this to set the profiling mode for the tests defined in the child class

getDataType

public DataType getDataType()

Override this to set the datatype of the tests defined in the child class

getDefaultFPDataType

public DataType getDefaultFPDataType()

isIntegrationTests

public static boolean isIntegrationTests()

Returns:

True if integration tests maven profile is enabled, false otherwise.

skipUnlessIntegrationTests

public static void skipUnlessIntegrationTests()

Call this as the first line of a test in order to skip that test, only when the integration tests maven profile is not enabled. This can be used to dynamically skip integration tests when the integration test profile is not enabled. Note that the integration test profile is not enabled by default - "integration-tests" profile