async def METHOD_NAME(subscription_state):
    tp1 = TopicPartition("topic", 0)
    tp2 = TopicPartition("topic", 1)
    subscription_state.assign_from_user({tp1})
    with pytest.raises(IllegalStateError):
        subscription_state._assigned_state(tp2)
    tp_state = subscription_state._assigned_state(tp1)
    assert tp_state is not None
    assert repr(tp_state) == (
        "TopicPartitionState<Status=PartitionStatus.AWAITING_RESET"
        " position=None>"
    )