Add some quick tests on the assume() logic to verify that it works as desired