Exemplo n.º 1
0
def test_calc_Z_MRIO(td_small_MRIO):
    pdt.assert_frame_equal(td_small_MRIO.Z,
                           calc_Z(td_small_MRIO.A, td_small_MRIO.x))
    # we also test the different methods to provide x:
    x_values = td_small_MRIO.x.values
    x_Tvalues = td_small_MRIO.x.T.values
    x_series = pd.Series(td_small_MRIO.x.iloc[:, 0])
    pdt.assert_frame_equal(td_small_MRIO.Z, calc_Z(td_small_MRIO.A, x_series))
    pdt.assert_frame_equal(td_small_MRIO.Z, calc_Z(td_small_MRIO.A, x_values))
    pdt.assert_frame_equal(td_small_MRIO.Z, calc_Z(td_small_MRIO.A, x_Tvalues))
Exemplo n.º 2
0
def test_calc_Z_arr(td_IO_Data_Miller):
    npt.assert_array_equal(
        td_IO_Data_Miller.Z_arr,
        calc_Z(td_IO_Data_Miller.A_arr, td_IO_Data_Miller.x_arr))
Exemplo n.º 3
0
def test_calc_Z_df(td_IO_Data_Miller):
    pdt.assert_frame_equal(
        td_IO_Data_Miller.Z_df,
        calc_Z(td_IO_Data_Miller.A_df, td_IO_Data_Miller.x_df))
Exemplo n.º 4
0
def test_calc_Z_arr(td_IO_Data_Miller):
    npt.assert_array_equal(
            td_IO_Data_Miller.Z_arr,
            calc_Z(td_IO_Data_Miller.A_arr, td_IO_Data_Miller.x_arr)
            )
Exemplo n.º 5
0
def test_calc_Z_df(td_IO_Data_Miller):
    pdt.assert_frame_equal(
            td_IO_Data_Miller.Z_df,
            calc_Z(td_IO_Data_Miller.A_df, td_IO_Data_Miller.x_df)
            )