def test_calc_split_raise():
dsi = DatasetIndex(5)
with pytest.raises(ValueError):
dsi.calc_split(shares=[0.5, 0.5, 0.5])
with pytest.raises(ValueError):
dsi.calc_split(shares=[0.5, 0.5, 0.5, 0.5])
with pytest.raises(ValueError):
DatasetIndex(2).calc_split(shares=[0.5, 0.5, 0.5])
def test_calc_split_correctness_3():
""" If 'shares' contains 3 elements, then validation subset is non-empty. """
dsi = DatasetIndex(5)
_, _, valid_share = dsi.calc_split(shares=[0.5, 0.5, 0])
assert valid_share == 0
def test_calc_split_correctness_2():
""" If 'shares' contains 2 elements, validation subset is empty. """
dsi = DatasetIndex(5)
left = dsi.calc_split(shares=[0.4, 0.6])
right = (2, 3, 0)
assert left == right
def test_calc_split_correctness_1():
dsi = DatasetIndex(5)
assert sum(dsi.calc_split()) == 5
