def test_create_batch_pos_false_int(): """ When 'pos' is False, method returns the same, as its first argument. """ dsi = DatasetIndex(5) left = dsi.create_batch(range(3), pos=False).index assert (left == range(3)).all()
def test_create_batch_pos_true_str(): """ When 'pos' is True, method creates new batch by specified positions. """ dsi = DatasetIndex(['a', 'b', 'c', 'd', 'e']) left = dsi.create_batch(range(5), pos=True).index assert (left == ['a', 'b', 'c', 'd', 'e']).all()
def test_create_batch_pos_false_str(): """ When 'pos' is False, method returns the same, as its first argument. """ dsi = DatasetIndex(['a', 'b', 'c', 'd', 'e']) left = dsi.create_batch(['a', 'e'], pos=False).index assert (left == ['a', 'e']).all()
def test_create_batch_pos_true_list(): """ When 'pos' is True, method creates new batch by specified positions. """ dsi = DatasetIndex(range(10, 20, 2)) left = dsi.create_batch(range(5), pos=True).index assert (left == range(10, 20, 2)).all()