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
def getCensusLoaderForSeq(self, table_ids): cl = CensusLoader(table_ids, []) cl.open_snatnl = mock.MagicMock(name='open_snatnl') cl.open_snatnl.return_value = StringIO.StringIO(sample_sequence) return cl
def test_to_title_1(self): in1 = 'THIS IS A YELLING TITLE' expected = 'This is a Yelling Title' self.assertEqual(expected, CensusLoader.to_title(in1))
def handle(self, *args, **options): cl = CensusLoader(table_ids=TABLES, states=[]) self.stdout.write(cl.model_declaration())
def handle(self, *args, **options): logging.basicConfig( level=logging.INFO, format=" %(levelname)s %(name)s: %(message)s") cl = CensusLoader(table_ids=TABLES, states=STATES) cl.import_data()