get rid of mocks in favor of python mock.