Add py.test as test runner