def test_one_row_group(self):
ONE_ROW = dict(
company='Foo',
scraper_company='Foo',
scraper_id='sr.campaign.bar')
insert_row(self.output_db, 'scraper_company_map', ONE_ROW)
self.assertEqual(
list(select_groups(
self.output_db, 'scraper_company_map', ['company'])),
[(('Foo',), [ONE_ROW])])
def test_two_row_group(self):
TWO_ROWS = [
dict(company='Foo',
scraper_company='Foo',
scraper_id='sr.campaign.bar'),
dict(company='Foo',
scraper_company='Foo & Co.',
scraper_id='sr.campaign.qux'),
]
insert_rows(self.output_db, 'scraper_company_map', TWO_ROWS)
groups = [(key, sorted_rows(rows))
for key, rows in select_groups(
self.output_db, 'scraper_company_map', ['company'])]
self.assertEqual(groups,
[(('Foo',), TWO_ROWS)])
def test_empty(self):
self.assertEqual(
list(select_groups(
self.output_db, 'scraper_company_map', ['company'])),
[])
