re-enable some tests
This commit is contained in:
Родитель
223bb0a6c2
Коммит
546af818d2
|
@ -19,8 +19,8 @@ import numpy as np
|
|||
# Testing inputs
|
||||
@pytest.mark.parametrize("left_operand, right_operand", [
|
||||
([30], [10]),
|
||||
#([[30]], [[10]]),
|
||||
#([[1.5,2.1]], [[10,20]]),
|
||||
([[30]], [[10]]),
|
||||
([[1.5,2.1]], [[10,20]]),
|
||||
#TODO: enable once all branches are merged to master
|
||||
#([5], [[30,40], [1,2]]),
|
||||
#Adding two 3x2 inputs of sequence length 1
|
||||
|
@ -44,15 +44,15 @@ def test_op_plus(left_operand, right_operand, device_id, precision):
|
|||
precision=precision, clean_up=False, backward_pass=False)
|
||||
|
||||
right_as_input = left_operand + b
|
||||
#unittest_helper(right_as_input, expected, device_id=device_id,
|
||||
# precision=precision, clean_up=True, backward_pass=False)
|
||||
unittest_helper(right_as_input, expected, device_id=device_id,
|
||||
precision=precision, clean_up=True, backward_pass=False)
|
||||
|
||||
#Backward pass test
|
||||
#==================
|
||||
#the expected results for the backward pass is all ones
|
||||
expected = [[[np.ones_like(x) for x in left_operand]]]
|
||||
#unittest_helper(left_as_input, expected, device_id=device_id,
|
||||
# precision=precision, clean_up=True, backward_pass=True, input_node=a)
|
||||
#unittest_helper(right_as_input, expected, device_id=device_id,
|
||||
# precision=precision, clean_up=True, backward_pass=True, input_node=b)
|
||||
unittest_helper(left_as_input, expected, device_id=device_id,
|
||||
precision=precision, clean_up=True, backward_pass=True, input_node=a)
|
||||
unittest_helper(right_as_input, expected, device_id=device_id,
|
||||
precision=precision, clean_up=True, backward_pass=True, input_node=b)
|
||||
|
Загрузка…
Ссылка в новой задаче