Skip to content

Commit 9fc5dea

Browse files
committed
Add another StoreOut test.
1 parent 1062373 commit 9fc5dea

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

DeepFried2/tests/containers/test_StoreOut.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,9 @@ def test(self):
1515
Y = net.forward(X)
1616
np.testing.assert_array_equal(net.out, Y)
1717

18+
Y = net.forward(X*2)
19+
np.testing.assert_array_equal(net.out, Y)
20+
1821
net.evaluate()
1922
X = np.array([[10,20],[30,40]], dtype=df.floatX)
2023
Y = net.forward(X)

0 commit comments

Comments
 (0)