Skip to content

Commit a98d2bb

Browse files
committed
polish unittest
1 parent e56d536 commit a98d2bb

File tree

1 file changed

+4
-1
lines changed
  • python/paddle/fluid/tests/unittests/dygraph_to_static

1 file changed

+4
-1
lines changed

python/paddle/fluid/tests/unittests/dygraph_to_static/test_slice.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,6 +187,7 @@ def test_compare_paddle_strided_slice_with_numpy(self):
187187
stride1 = -2
188188
sl = paddle.strided_slice(
189189
pt, axes=[0, ], starts=[s1, ], ends=[e1, ], strides=[stride1, ])
190+
190191
self.assertTrue(array[s1:e1:stride1], sl)
191192

192193
array = np.arange(6 * 6).reshape((6, 6))
@@ -197,7 +198,9 @@ def test_compare_paddle_strided_slice_with_numpy(self):
197198
sl = paddle.strided_slice(
198199
pt, axes=[0, 1], starts=s2, ends=e2, strides=stride2)
199200

200-
array[s2[0]:e2[0]:stride2[0], s2[1]:e2[1]:stride2[1]]
201+
self.assertTrue(
202+
np.array_equal(sl.numpy(), array[s2[0]:e2[0]:stride2[0], s2[1]:e2[
203+
1]:stride2[1]]))
201204

202205

203206
if __name__ == '__main__':

0 commit comments

Comments
 (0)