update unit tests for the new input checks

I hope the test cases are taken right, please check before merging.