Replace assert with should for assertions in unit tests