def test_step_entangler_compute_fragment_size_raises_ValueError_if_document_size_is_negative(
):
    with pytest.raises(ValueError) as excinfo:
        StepEntangler.compute_fragment_size(-1, 1)
    assert str(
        excinfo.value
    ) == "document_size argument must be an integer greater or equal to 0"
def test_step_entangler_compute_fragment_size():
    assert StepEntangler.compute_fragment_size(100, 1) == 100
    assert StepEntangler.compute_fragment_size(101, 1) == 102
    assert StepEntangler.compute_fragment_size(100, 2) == 50
    assert StepEntangler.compute_fragment_size(100, 3) == 34
def test_step_entangler_compute_fragment_size_raises_ValueError_if_fragments_is_equal_to_0(
):
    with pytest.raises(ValueError) as excinfo:
        StepEntangler.compute_fragment_size(1, 0)
    assert str(excinfo.value
               ) == "fragments argument must be an integer greater than 0"