def test_batch(self): self.assertSequenceEqual( utils.batch(2, range(10)), [range(0, 2), range(2, 4), range(4, 6), range(6, 8), range(8, 10)]) self.assertEqual(utils.batch(3, [429, 5, 2, 14, 42, 132, 1, 1]), [[429, 5, 2], [14, 42, 132], [1, 1]]) self.assertEqual( utils.batch(4, range(10)), [range(0, 4), range(4, 8), range(8, 10)])
def matrix(ncol, *args): return np.array(utils.batch(ncol, args))
def testbatch(self): self.assertEqual(utils.batch(3, [429, 5, 2, 14, 42, 132, 1, 1]), [[429, 5, 2], [14, 42, 132], [1, 1]]) self.assertEqual(utils.batch(4, range(10)), [range(0, 4), range(4, 8), range(8, 10)])