def test_matched_ids(self): l = matched_ids(all_edges(get_data())) self.assertEquals(len(list(l)), 2)
def test_get_data(self): expected_count = DirectorFiling.objects.count() df = get_data() self.assertEquals(df.shape[0], expected_count)