def METHOD_NAME(projects_rebalancing_model):
    classes = [
        RebalancedItem(id=1, count=2.0),
        RebalancedItem(id=2, count=10.0),
        RebalancedItem(id=3, count=10.0),
        RebalancedItem(id=4, count=10.0),
    ]
    expected_classes = [
        RebalancedItem(
            id=1,
            count=2.0,
            new_sample_rate=1.0,
        ),
        RebalancedItem(
            id=2,
            count=10.0,
            new_sample_rate=0.2,
        ),
        RebalancedItem(
            id=3,
            count=10.0,
            new_sample_rate=0.2,
        ),
        RebalancedItem(
            id=4,
            count=10.0,
            new_sample_rate=0.2,
        ),
    ]
    assert (
        sorted(
            projects_rebalancing_model.run(
                ProjectsRebalancingInput(classes=classes, sample_rate=0.25)
            ),
            key=attrgetter("id"),
        )
        == expected_classes
    )