Beispiel #1
0
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
Beispiel #2
0
def test_it_parses_nycdb_files():
    assert 'hpd_registrations_grouped_by_bbl' in parse_nycdb_created_tables(
        ['hpd_registrations/registrations_grouped_by_bbl.sql'])