def get_dataset_tables() -> List[TableInfo]: result: List[TableInfo] = [] for dataset_name, info in nycdb.dataset.datasets().items(): for schema in list_wrap(info['schema']): result.append(TableInfo(name=schema['table_name'], dataset=dataset_name)) result.extend([ TableInfo(name=name, dataset=dataset_name) for name in parse_nycdb_created_tables(info.get('sql', [])) ]) return result
def test_it_parses_nycdb_files(): assert 'hpd_registrations_grouped_by_bbl' in parse_nycdb_created_tables( ['hpd_registrations/registrations_grouped_by_bbl.sql'])