Use `assert` funciton in tests.

Closes #52.