Adding basic testing framework

Like the docstring on the first test, would love one here too.