Testing framework

Shouldn't we use `print` as a function since we're aiming for Python 3-compatibility. Or better, just an assert.