better mocks for python