Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Attributes | List of all members
common_utils.TestCase Class Reference
Inheritance diagram for common_utils.TestCase:
expecttest.TestCase common_nn.NNTestCase test_c10d.FileStoreTest test_c10d.MultiProcessTestCase test_c10d.PrefixFileStoreTest test_c10d.PrefixTCPStoreTest test_c10d.ProcessGroupNCCLTest test_c10d.RendezvousEnvTest test_c10d.RendezvousFileTest test_c10d.RendezvousTCPTest test_c10d.RendezvousTest test_c10d.TCPStoreTest test_cuda_primary_ctx.TestCudaPrimaryCtx test_distributed.TestDistBackend test_distributed.TestMPI test_distributions.TestAgainstScipy test_distributions.TestConstraintRegistry test_distributions.TestConstraints test_distributions.TestDistributions test_distributions.TestDistributionShapes test_distributions.TestJit test_distributions.TestKL test_distributions.TestLazyLogitsInitialization test_distributions.TestNumericalStability test_distributions.TestRsample test_distributions.TestTransforms test_distributions.TestValidation test_indexing.NumpyTests test_indexing.TestIndexing test_jit.JitTestCase test_jit.TestBatched test_nccl.TestNCCL test_nn.PackedSequenceTest test_nn.TestNNInit test_optim.TestLRScheduler test_optim.TestOptim test_sparse.TestSparse test_sparse.TestSparseOneOff test_thd_distributed.TestDistBackend test_thd_distributed.TestMPI test_torch.TestTorch test_type_hints.TestTypeHints test_type_info.TestDTypeInfo test_utils.TestBottleneck test_utils.TestCheckpoint test_utils.TestCollectEnv test_utils.TestDataLoader test_utils.TestFFI test_utils.TestHub test_utils.TestONNXUtils

Public Member Functions

def __init__ (self, method_name='runTest')
 
def assertLeaksNoCudaTensors (self, name=None)
 
def wrap_with_cuda_memory_check (self, method)
 
def setUp (self)
 
def assertTensorsSlowEqual (self, x, y, prec=None, message='')
 
def genSparseTensor (self, size, sparse_dim, nnz, is_uncoalesced, device='cpu')
 
def safeToDense (self, t)
 
def safeCoalesce (self, t)
 
def assertEqual (self, x, y, prec=None, message='', allow_inf=False)
 
def assertAlmostEqual (self, x, y, places=None, msg=None, delta=None, allow_inf=None)
 
def assertNotEqual (self, x, y, prec=None, message='')
 
def assertObjectIn (self, obj, iterable)
 
def assertExpectedRaises (self, exc_type, callable, args, kwargs)
 
def assertWarns (self, callable, msg='')
 
def assertWarnsRegex (self, callable, regex, msg='')
 
def assertExpected (self, s, subname=None)
 
- Public Member Functions inherited from expecttest.TestCase
def assertExpectedInline (self, actual, expect, skip=0)
 

Static Public Attributes

int precision = 1
 
 maxDiff = None
 
 assertRegex = unittest.TestCase.assertRegexpMatches
 
 assertRaisesRegex = unittest.TestCase.assertRaisesRegexp
 
- Static Public Attributes inherited from expecttest.TestCase
bool longMessage = True
 

Detailed Description

Definition at line 265 of file common_utils.py.


The documentation for this class was generated from the following file: