def test_each_yielded_item_is_a_dictionary(self): result = scraper.extract_raw_data(generate_fake_rows()) for a in result: self.assertTrue(type(a) is dict)
def test_keys_for_dict_are_the_labels(self): row = scraper.extract_raw_data(generate_fake_rows()).next() expected = sorted(scraper.LABELS) actual = sorted(row.keys()) self.assertEqual(expected, actual)
def test_returns_a_generator(self): result = scraper.extract_raw_data(generate_fake_rows()) self.assertTrue(type(result) is types.GeneratorType)