def getCensusLoaderForGeo(self, fake_geo): self.assertEqual(1269, len(fake_geo)) cl = CensusLoader([], ['xx']) cl.open_geo = mock.MagicMock(name='open_geo') cl.open_geo.return_value = StringIO.StringIO(fake_geo) cl.add_geo_record = mock.MagicMock(name='add_geo_record') cl.add_geo_record.return_value = True return cl