For #206 Adding a couple sanity checks in tests