Skip to content
Permalink
Browse files

test: simplify yaml naming

  • Loading branch information...
hanxiao committed Aug 6, 2019
1 parent 04576bb commit 23a83a40c3b5ba094c155e67717da272527888a4
Showing with 5 additions and 1 deletion.
  1. +5 −1 tests/test_batching.py
@@ -104,4 +104,8 @@ def _do_mini_batch2(_, y):
return y

self.assertEqual(_do_mini_batch(None, x), [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9]])
self.assertEqual(_do_mini_batch2(None, x), [0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
self.assertEqual(_do_mini_batch2(self, x), [0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

self.batch_size = 4

self.assertEqual(_do_mini_batch2(self, x), [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9]])

0 comments on commit 23a83a4

Please sign in to comment.
You can’t perform that action at this time.