Add sanity check unit test