Added backwards compatibility with old numpy lacking argpartition
Add on_unused_input=warn to beamsearch test as it may not be set by deafult.