Add assert test