This commit is contained in:
jeanfad 2016-04-08 10:21:33 +02:00
Родитель 223bb0a6c2
Коммит 546af818d2
1 изменённых файлов: 8 добавлений и 8 удалений

Просмотреть файл

@ -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)