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_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)
