def test_save_related(): s = Scraper('jurisdiction', '2013', '/tmp/') p = Person('Michael Jordan') p.add_source('http://example.com') o = Organization('Chicago Bulls') o.add_source('http://example.com') p._related.append(o) with mock.patch('json.dump') as json_dump: s.save_object(p) assert_equal(json_dump.mock_calls, [ mock.call(p.as_dict(), mock.ANY, cls=mock.ANY), mock.call(o.as_dict(), mock.ANY, cls=mock.ANY) ])