def test_discard_null_judgment(self):
# this tests issue #22
insert_row(self.scratch_db, 'rating', dict(
scraper_id='sr.campaign.qux',
campaign_id='qux',
company='Foo & Co.',
brand='',
scope=''))
build_rating_table(self.output_db, self.scratch_db)
self.assertEqual(select_all(self.output_db, 'rating'), [])
def test_map_brand(self):
insert_row(self.scratch_db, 'rating', dict(
scraper_id='sr.campaign.qux',
campaign_id='qux',
company='Foo & Co.',
brand='BAR™',
judgment=-1))
build_rating_table(self.output_db, self.scratch_db)
self.assertEqual(
[strip_null(row) for row in select_all(self.output_db, 'rating')],
[dict(campaign_id='qux',
company='Foo',
brand='Bar',
scope='',
judgment=-1)])
def test_infer_judgment_from_grade(self):
# this tests issue #23
insert_row(self.scratch_db, 'rating', dict(
scraper_id='sr.campaign.qux',
campaign_id='qux',
company='Foo & Co.',
brand='',
scope='',
grade='B'))
build_rating_table(self.output_db, self.scratch_db)
self.assertEqual(
[strip_null(row) for row in select_all(self.output_db, 'rating')],
[dict(campaign_id='qux',
company='Foo',
brand='',
scope='',
grade='B',
judgment=1)])
