updated tests to use assert