def test_get_batch_batch_size_greater_than_actual(self): executions_queue = self.get_queue() qsize = executions_queue.qsize() self.assertTrue(qsize > 0) dumper = Dumper(queue=executions_queue, batch_size=2 * qsize, export_dir='/tmp') batch = dumper._get_batch() self.assertEqual(len(batch), qsize)
def test_get_batch_batch_size_lesser_than_actual(self): executions_queue = self.get_queue() qsize = executions_queue.qsize() self.assertTrue(qsize > 0) expected_batch_size = int(qsize / 2) dumper = Dumper(queue=executions_queue, batch_size=expected_batch_size, export_dir="/tmp") batch = dumper._get_batch() self.assertEqual(len(batch), expected_batch_size)