def test_get_or_construct_data_CN1(): # copied on the one in cn_test df = get_comptes_nationaux_data(2013) values_CN1_target = read_CN1(2013) variables_CN1 = generate_CN1_variables(2013) values_CN1, formulas_CN1 = get_or_construct_data(df, variables_CN1, range(1949, 2014)) assert_frame_equal(values_CN1, values_CN1_target)
def test_compare_sheets_to_IPP(): # WIP. for now it only generates the first two sheets, to be compared to Agrégats IPP df = get_tidy_data(2012) variables_CN1 = generate_CN1_variables(2012) variables_CN2 = generate_CN2_variables(2012) values_CN1_2012, formulas_CN1 = get_or_construct_data(df, variables_CN1, range(1949, 2013)) values_CN2_2012, formulas_CN2 = get_or_construct_data(df, variables_CN2, range(1949, 2013)) return values_CN1_2012, values_CN2_2012
def test_CN1(year): df = get_tidy_data(year) values_CN1_target = read_CN1(year) variables_CN1 = generate_CN1_variables(year) values_CN1, formulas_CN1 = get_or_construct_data(df, variables_CN1, range(1949, year + 1)) assert_frame_equal(values_CN1, values_CN1_target)
def create_and_save_CN1(year): # NB : should no longer be executed - already saved file_path = (os.path.join(tests_data, 'values_CN1_{}.h5'.format(year))) assert not os.path.exists(file_path), 'CN1 sheet with {} data already generated and saved'.format(year) df = get_tidy_data(year) variables_CN1 = generate_CN1_variables(year) values_CN1, formulas_CN1 = get_or_construct_data(df, variables_CN1, range(1949, year + 1)) values_CN1.to_hdf(file_path, 'CN1')
def test_CN1(year): df = get_tidy_data(year) values_CN1_target = read_CN1(year) variables_CN1 = generate_CN1_variables(year) values_CN1, formulas_CN1 = get_or_construct_data(df, variables_CN1, range(1949, year + 1)) assert_frame_equal(values_CN1, values_CN1_target)
def test_get_or_construct_data_CN1(): # copied on the one in cn_test df = get_comptes_nationaux_data(2013) values_CN1_target = read_CN1(2013) variables_CN1 = generate_CN1_variables(2013) values_CN1, formulas_CN1 = get_or_construct_data(df, variables_CN1, range(1949, 2014)) print values_CN1.columns print values_CN1_target.columns assert_frame_equal(values_CN1, values_CN1_target)
def test_compare_sheets_to_IPP( ): # WIP. for now it only generates the first two sheets, to be compared to Agrégats IPP df = get_tidy_data(2012) variables_CN1 = generate_CN1_variables(2012) variables_CN2 = generate_CN2_variables(2012) values_CN1_2012, formulas_CN1 = get_or_construct_data( df, variables_CN1, range(1949, 2013)) values_CN2_2012, formulas_CN2 = get_or_construct_data( df, variables_CN2, range(1949, 2013)) return values_CN1_2012, values_CN2_2012
def create_and_save_CN1( year): # NB : should no longer be executed - already saved file_path = (os.path.join(tests_data, 'values_CN1_{}.h5'.format(year))) assert not os.path.exists( file_path ), 'CN1 sheet with {} data already generated and saved'.format(year) df = get_tidy_data(year) variables_CN1 = generate_CN1_variables(year) values_CN1, formulas_CN1 = get_or_construct_data(df, variables_CN1, range(1949, year + 1)) values_CN1.to_hdf(file_path, 'CN1')
def generate_CN1(year): df = get_tidy_data(year) variables_CN1 = generate_CN1_variables(year) values_CN1, formulas_CN1 = get_or_construct_data(df, variables_CN1, range(1949, year + 1)) return values_CN1, formulas_CN1
def generate_CN1(year): df = get_tidy_data(year) variables_CN1 = generate_CN1_variables(year) values_CN1, formulas_CN1 = get_or_construct_data(df, variables_CN1, range(1949, year + 1)) return values_CN1, formulas_CN1