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