Skip to content
This repository has been archived by the owner on Feb 22, 2020. It is now read-only.

Commit

Permalink
test: simplify yaml naming
Browse files Browse the repository at this point in the history
  • Loading branch information
hanhxiao committed Aug 6, 2019
1 parent 04576bb commit 23a83a4
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion tests/test_batching.py
Expand Up @@ -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.