update unit testing, and remove the unused assert functions