add py.test tests

:)