def test_column_headers(database):
    expected = {
        'row_number', 'allocation_transfer_agency', 'agency_identifier',
        'beginning_period_of_availa', 'ending_period_of_availabil',
        'availability_type_code', 'main_account_code', 'sub_account_code',
    }
    assert expected == set(query_columns(_FILE, database))

def test_column_headers(database):
    expected_subset = {'row_number', 'beginning_period_of_availa', 'ending_period_of_availabil',
                       'agency_identifier', 'allocation_transfer_agency', 'availability_type_code',
                       'main_account_code', 'sub_account_code', 'object_class', 'program_activity_code',
                       'by_direct_reimbursable_fun'}
    actual = set(query_columns(_FILE, database))
    assert (actual & expected_subset) == expected_subset

def test_column_headers(database):
    expected_subset = {
        'row_number', 'allocation_transfer_agency', 'agency_identifier', 'beginning_period_of_availa',
        'ending_period_of_availabil', 'availability_type_code', 'main_account_code', 'sub_account_code',
        'gross_outlay_amount_by_tas_cpe', 'gross_outlay_amount_by_pro_cpe_sum'
    }
    actual = set(query_columns(_FILE, database))
    assert (actual & expected_subset) == expected_subset

def test_column_headers(database, sql_file):
    expected_subset = {
        "row_number", "allocation_transfer_agency", "agency_identifier",
        "beginning_period_of_availa", "ending_period_of_availabil",
        "availability_type_code", "main_account_code", "sub_account_code"
    }

    actual = set(query_columns(sql_file, database))
    assert expected_subset == actual

def test_column_headers(database):
    expected_subset = {'row_number', 'tas', 'program_activity_code', 'ussgl480100_undelivered_or_fyb_sum_c',
                       'ussgl480100_undelivered_or_cpe_sum_c', 'ussgl483100_undelivered_or_cpe_sum_c',
                       'ussgl488100_upward_adjustm_cpe_sum_c', 'obligations_undelivered_or_fyb_sum_c',
                       'obligations_undelivered_or_cpe_sum_c', 'ussgl490100_delivered_orde_fyb_sum_c',
                       'ussgl490100_delivered_orde_cpe_sum_c', 'ussgl493100_delivered_orde_cpe_sum_c',
                       'ussgl498100_upward_adjustm_cpe_sum_c', 'obligations_delivered_orde_fyb_sum_c',
                       'obligations_delivered_orde_cpe_sum_c', 'ussgl480200_undelivered_or_fyb_sum_c',
                       'ussgl480200_undelivered_or_cpe_sum_c', 'ussgl483200_undelivered_or_cpe_sum_c',
                       'ussgl488200_upward_adjustm_cpe_sum_c', 'gross_outlays_undelivered_fyb_sum_c',
                       'gross_outlays_undelivered_cpe_sum_c', 'ussgl490200_delivered_orde_cpe_sum_c',
                       'ussgl490800_authority_outl_fyb_sum_c', 'ussgl490800_authority_outl_cpe_sum_c',
                       'ussgl498200_upward_adjustm_cpe_sum_c', 'gross_outlays_delivered_or_fyb_sum_c',
                       'gross_outlays_delivered_or_cpe_sum_c', 'gross_outlay_amount_by_awa_fyb_sum_c',
                       'gross_outlay_amount_by_awa_cpe_sum_c', 'obligations_incurred_byawa_cpe_sum_c',
                       'ussgl487100_downward_adjus_cpe_sum_c', 'ussgl497100_downward_adjus_cpe_sum_c',
                       'ussgl487200_downward_adjus_cpe_sum_c', 'ussgl497200_downward_adjus_cpe_sum_c',
                       'deobligations_recov_by_awa_cpe_sum_c', 'ussgl480100_undelivered_or_fyb_sum_b',
                       'ussgl480100_undelivered_or_cpe_sum_b', 'ussgl483100_undelivered_or_cpe_sum_b',
                       'ussgl488100_upward_adjustm_cpe_sum_b', 'obligations_undelivered_or_fyb_sum_b',
                       'obligations_undelivered_or_cpe_sum_b', 'ussgl490100_delivered_orde_fyb_sum_b',
                       'ussgl490100_delivered_orde_cpe_sum_b', 'ussgl493100_delivered_orde_cpe_sum_b',
                       'ussgl498100_upward_adjustm_cpe_sum_b', 'obligations_delivered_orde_fyb_sum_b',
                       'obligations_delivered_orde_cpe_sum_b', 'ussgl480200_undelivered_or_fyb_sum_b',
                       'ussgl480200_undelivered_or_cpe_sum_b', 'ussgl483200_undelivered_or_cpe_sum_b',
                       'ussgl488200_upward_adjustm_cpe_sum_b', 'gross_outlays_undelivered_fyb_sum_b',
                       'gross_outlays_undelivered_cpe_sum_b', 'ussgl490200_delivered_orde_cpe_sum_b',
                       'ussgl490800_authority_outl_fyb_sum_b', 'ussgl490800_authority_outl_cpe_sum_b',
                       'ussgl498200_upward_adjustm_cpe_sum_b', 'gross_outlays_delivered_or_fyb_sum_b',
                       'gross_outlays_delivered_or_cpe_sum_b', 'gross_outlay_amount_by_pro_fyb_sum_b',
                       'gross_outlay_amount_by_pro_cpe_sum_b', 'obligations_incurred_by_pr_cpe_sum_b',
                       'ussgl487100_downward_adjus_cpe_sum_b', 'ussgl497100_downward_adjus_cpe_sum_b',
                       'ussgl487200_downward_adjus_cpe_sum_b', 'ussgl497200_downward_adjus_cpe_sum_b',
                       'deobligations_recov_by_pro_cpe_sum_b'}
    actual = set(query_columns(_FILE, database))
    assert (actual & expected_subset) == expected_subset

def test_column_headers(database):
    expected_subset = {"row_number", "action_type", "record_type"}
    actual = set(query_columns(_FILE, database))
    assert expected_subset == actual

def test_column_headers(database):
    expected_subset = {'row_number', 'deobligations_recov_by_pro_cpe', 'ussgl487100_downward_adjus_cpe',
                       'ussgl487200_downward_adjus_cpe', 'ussgl497100_downward_adjus_cpe',
                       'ussgl497200_downward_adjus_cpe'}
    actual = set(query_columns(_FILE, database))
    assert (actual & expected_subset) == expected_subset

def test_column_headers(database):
    expected_subset = {"row_number", "awarding_office_code"}
    actual = set(query_columns(_FILE, database))
    assert expected_subset == actual

def test_column_headers(database):
    expected_subset = {'row_number', 'object_class'}
    actual = set(query_columns(_FILE, database))
    assert (actual & expected_subset) == expected_subset

def test_column_headers(database):
    expected_subset = {'row_number', 'obligations_delivered_orde_cpe', 'ussgl490100_delivered_orde_cpe',
                       'ussgl498100_upward_adjustm_cpe'}
    actual = set(query_columns(_FILE, database))
    assert (actual & expected_subset) == expected_subset

def test_column_headers(database):
    expected_subset = {'row_number', 'piid', 'parent_award_id'}
    actual = set(query_columns(_FILE, database))
    assert (actual & expected_subset) == expected_subset

def test_column_headers(database):
    expected_subset = {'row_number', 'agency_identifier', 'main_account_code',
                       'program_activity_name', 'program_activity_code'}
    actual = set(query_columns(_FILE, database))
    assert (actual & expected_subset) == expected_subset

def test_column_headers(database):
    expected_subset = {"row_number", "assistance_type", "action_date", "awardee_or_recipient_uniqu",
                       "business_types", "record_type"}
    actual = set(query_columns(_FILE, database))
    assert expected_subset == actual

def test_column_headers(database):
    expected_subset = {"row_number", "cfda_number", "action_date", "action_type", "correction_late_delete_ind"}
    actual = set(query_columns(_FILE, database))
    assert expected_subset == actual

def test_column_headers(database):
    expected_subset = {"row_number", "legal_entity_country_code", "legal_entity_foreign_city", "record_type"}
    actual = set(query_columns(_FILE, database))
    assert expected_subset == actual

def test_column_headers(database):
    expected_subset = {"row_number", "transaction_obligated_amou_sum", "federal_action_obligation_sum",
                       "original_loan_subsidy_cost_sum"}
    actual = set(query_columns(_FILE, database))
    assert expected_subset <= actual

def test_column_headers(database):
    expected_subset = {'row_number', "fain", "uri", "piid"}
    actual = set(query_columns(_FILE, database))
    assert (actual & expected_subset) == expected_subset

def test_column_headers(database):
    expected_subset = {"row_number", "status_of_budgetary_resour_cpe", "obligations_incurred_total_cpe",
                       "unobligated_balance_cpe"}
    actual = set(query_columns(_FILE, database))
    assert expected_subset == actual