Instead of using a combination of TestCase.assertEqual + mock.call_args, you can just check with mock.assert_called_with:
http://www.voidspace.org.uk/python/mock/mock.html#mock.Mock.assert_called_with

Can you change this for the assertions in this pull request, so that the tests get more readable?