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