add assertions in unit tests