TestUtil Namespace Reference

Functions
	assertTrue (cond, msg)
	checkPadding (pad, exp_pad)
	checkActivation (act, exp_act)
	checkOpcode (opcode, exp_opcode)
	checkBuiltinOptionType (option, exp_option)

Function Documentation

assertTrue()

TestUtil.assertTrue	(		cond,
			msg
	)

Definition at line 21 of file TestUtil.py.

def assertTrue(cond, msg):
    assert cond, msg
 
 

Referenced by checkActivation(), checkBuiltinOptionType(), checkOpcode(), and checkPadding().

checkActivation()

TestUtil.checkActivation	(		act,
			exp_act
	)

Definition at line 34 of file TestUtil.py.

def checkActivation(act, exp_act):
    act_functions = {
        'relu': ActivationFunctionType.ActivationFunctionType.RELU,
        'relu6': ActivationFunctionType.ActivationFunctionType.RELU6,
        'relu_n1_to_1': ActivationFunctionType.ActivationFunctionType.RELU_N1_TO_1,
        'tanh': ActivationFunctionType.ActivationFunctionType.TANH,
        'none': ActivationFunctionType.ActivationFunctionType.NONE,
        'sign_bit': ActivationFunctionType.ActivationFunctionType.SIGN_BIT,
    }
 
    try:
        assertTrue(act_functions[act] == exp_act, "Activation function mismatches")
    except KeyError:
        raise SystemExit('Unsupported activation functions')
 
 

References assertTrue().

checkBuiltinOptionType()

TestUtil.checkBuiltinOptionType	(		option,
			exp_option
	)

Definition at line 55 of file TestUtil.py.

def checkBuiltinOptionType(option, exp_option):
    assertTrue(
        option == exp_option,
        "Built-in option type mismatches (" + str(option) + ", " + str(exp_option) + ")")

References assertTrue().

checkOpcode()

TestUtil.checkOpcode	(		opcode,
			exp_opcode
	)

Definition at line 50 of file TestUtil.py.

def checkOpcode(opcode, exp_opcode):
    assertTrue(opcode == exp_opcode,
               "Opcode mismatches (" + str(opcode) + ", " + str(exp_opcode) + ")")
 
 

References assertTrue().

checkPadding()

TestUtil.checkPadding	(		pad,
			exp_pad
	)

Definition at line 25 of file TestUtil.py.

def checkPadding(pad, exp_pad):
    if pad == 'SAME':
        assertTrue(exp_pad == Padding.Padding.SAME, "Padding mismatches")
    elif pad == 'VALID':
        assertTrue(exp_pad == Padding.Padding.VALID, "Padding mismatches")
    else:
        raise SystemExit('Unsupported padding')
 
 

References assertTrue().