fab8d83aa1d19d5e40383323a60666a886804c54,batchflow/tests/rebatch_test.py,,test_rebatch,#Any#Any#,18

Before Change

def test_rebatch(batch_size, rebatch_size):
    dataset_size = 60
    batch_shape = (dataset_size, 10)
    data = np.random.random(batch_shape)

    dataset = Dataset(index=dataset_size,
                      batch_class=MyBatch,
                      preloaded=data)

After Change

@pytest.mark.parametrize("batch_size, rebatch_size", PARAMETERS)
def test_rebatch(batch_size, rebatch_size):

    data = np.vstack([np.array([i, i]) for i in range(DATASET_SIZE)])
    dataset = Dataset(index=DATASET_SIZE,
                      batch_class=MyBatch,
                      preloaded=data)

Instances