Ejemplos de group_array_elements en Python

Ejemplo n.º 1

def test_equal_element_and_group_count():
    assert group_array_elements([1, 2, 3, 4], 4) == [[1], [2], [3], [4]]

Ejemplo n.º 2

def test_sample():
    assert group_array_elements([1, 2, 3, 4, 5], 3) == [[1, 2], [3, 4], [5]]

Ejemplo n.º 3

def test_invalid_element_count():
    with pytest.raises(Exception) as ex:
        group_array_elements([1, 2], 3)

    assert str(ex.value) == 'not enough elements to group'

Ejemplo n.º 4

def test_invalid_group_count():
    with pytest.raises(Exception) as ex:
        group_array_elements([], -1)

    assert str(ex.value) == 'group_count must be a positive integer'

Ejemplo n.º 5

def test_uneven():
    assert group_array_elements([1, 2, 3, 4, 5, 6, 7, 8, 9],
                                2) == [[1, 2, 3, 4, 5], [6, 7, 8, 9]]

Ejemplo n.º 6

def test_no_remainder():
    assert group_array_elements([1, 2, 3, 4, 5, 6, 7, 8, 9], 3) == [[1, 2, 3],
                                                                    [4, 5, 6],
                                                                    [7, 8, 9]]