class TestCapitolWords(unittest.TestCase): def setUp(self): self.phrases_kwargs = { 'entity_type': 'legislator', 'entity_value': 'L000551' } self.service = CapitolWords() def test__get_url(self): '''This is probably a bad idea and should be replaced.''' url = self.service._get_url(['phrases'], sunlight.config.API_KEY, **self.phrases_kwargs) expected_hostname = 'capitolwords.org' expected_path = '/api/1/phrases.json' expected_query = { 'apikey': sunlight.config.API_KEY } expected_query.update(self.phrases_kwargs) parsed_url = urlparse(url) parsed_query = dict(parse_qsl(parsed_url.query)) self.assertEqual(parsed_url.hostname, expected_hostname) self.assertEqual(parsed_url.path, expected_path) self.assertDictEqual(parsed_query, expected_query) def test_dates(self): results = self.service.dates('Obamacare') self.assertNotEqual(len(results), 0) def test_phrases_by_entity(self): results = self.service.phrases_by_entity('state', phrase='Obamacare') self.assertNotEqual(len(results), 0) def test_legislator_phrases(self): results = self.service.phrases( self.phrases_kwargs['entity_type'], self.phrases_kwargs['entity_value']) self.assertNotEqual(len(results), 0) def test_text(self): results = self.service.text('Christmas') self.assertNotEqual(len(results), 0) def test_text_title(self): results = self.service.text(title='Christmas') self.assertNotEqual(len(results), 0) def test_text_bad_request(self): with self.assertRaises(BadRequestException): self.service.text()
try: with open('capitolwords.csv') as f: reader = csv.reader(f) for i, r in enumerate(reader): pass return i - 1 except: return 0 if __name__ == "__main__": if len(sys.argv) > 2: outfile = sys.argv[1] else: outfile = "capitolwords.csv" cw = CapitolWords() start_date = '2013-01-03' end_date = '2014-12-16' sort = "date asc" results = cw.text(phrase=" ", start_date=start_date, sort=sort) if len(results) > 1: headers = results[0].keys() else: print("Not found") sys.exit() nrow = csv_rowcount(outfile) print nrow if nrow > 0: start = nrow f = open(outfile, "ab")
try: with open('capitolwords.csv') as f: reader = csv.reader(f) for i, r in enumerate(reader): pass return i - 1 except: return 0 if __name__ == "__main__": if len(sys.argv) > 2: outfile = sys.argv[1] else: outfile = "capitolwords.csv" cw = CapitolWords() start_date = '2013-01-03' end_date = '2015-01-03' #end_date = datetime.today().strftime('%Y-%m-%d') sort = "date asc" results = cw.text(phrase=" ", start_date=start_date, end_date=end_date, sort=sort) if len(results) > 1: headers = results[0].keys() else: print("Not found") sys.exit() nrow = csv_rowcount(outfile) print nrow if nrow > 0: start = nrow
def setUp(self): self.phrases_kwargs = { 'entity_type': 'legislator', 'entity_value': 'L000551' } self.service = CapitolWords()