Function test_to_data_client_group

tests/test_Model.py:706–725

Tests that creating a small model with a client group results in a correct ProblemData instance that has the appropriate group memberships set up.

()

def test_to_data_client_group():
    """
    Tests that creating a small model with a client group results in a correct
    ProblemData instance that has the appropriate group memberships set up.
    """
    m = Model()
    m.add_depot(1, 1)
    m.add_vehicle_type()

    group = m.add_client_group()
    m.add_client(1, 1, required=False, group=group)
    m.add_client(2, 2, required=False, group=group)

    # Generate the data instance. There should be a single client group, and
    # the first two clients should be members of that group.
    data = m.data()
    assert_equal(data.num_groups, 1)

    group = data.group(0)
    assert_equal(group.clients, [1, 2])


def test_raises_mutually_exclusive_client_group_required_client():