unit tests

added an assert statement