test: simplify yaml naming

gnes-ai · Aug 6, 2019 · 23a83a4 · 23a83a4
1 parent 04576bb
commit 23a83a4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_batching.py b/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]])