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'])), [])