This also updates the test.sh script to use an assert. Signed-off-by: Jeffery To <jeffery.to@gmail.com>