async def METHOD_NAME(subscription_state):
    assert subscription_state.assigned_partitions() == set()
    subscription_state.subscribe(topics={"tp1"})
    assert subscription_state.assigned_partitions() == set()
    assignment = {TopicPartition("tp1", 0)}
    subscription_state.assign_from_subscribed(assignment)
    assert subscription_state.assigned_partitions() == assignment