6fc1a9f5d2c9f57f22b17308bffe9dfe50c2c29f,batchflow/tests/datasetindex_test.py,,test_next_batch_stopiter_pass,#Any#,130

Before Change

    dsi = index("small")
    dsi.reset_iter()
    dsi.next_batch(2, n_epochs=None)
    dsi.next_batch(2, n_epochs=None)

def test_next_batch_stopiter_raise(index):
     Iteration is blocked after end of DatasetIndex. 
    dsi = index("small")

After Change

    left = set()
    right = set(dsi.index)
    for _ in range(SIZE**2):
        n_b = dsi.next_batch(1, n_epochs=None)
        left = left | set(n_b.index)
    assert left == right

def test_next_batch_drop_last_false_1(all_indices):
     When "drop_last" is False "next_batch" should cycle through index. 

Instances