Change these null checks to be asserts
I am not super happy with this change, but hopefully it will make this unit test consistent
More unit tests