def test_calc_A_MRIO(td_small_MRIO): pdt.assert_frame_equal(td_small_MRIO.A, calc_A(td_small_MRIO.Z, 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.A, calc_A(td_small_MRIO.Z, x_series)) pdt.assert_frame_equal(td_small_MRIO.A, calc_A(td_small_MRIO.Z, x_values)) pdt.assert_frame_equal(td_small_MRIO.A, calc_A(td_small_MRIO.Z, x_Tvalues))
def test_calc_A_MRIO(td_small_MRIO): pdt.assert_frame_equal(td_small_MRIO.A, calc_A(td_small_MRIO.Z, td_small_MRIO.x))
def test_calc_A_arr(td_IO_Data_Miller): npt.assert_array_equal( td_IO_Data_Miller.A_arr, calc_A(td_IO_Data_Miller.Z_arr, td_IO_Data_Miller.x_arr))
def test_calc_A_df(td_IO_Data_Miller): pdt.assert_frame_equal( td_IO_Data_Miller.A_df, calc_A(td_IO_Data_Miller.Z_df, td_IO_Data_Miller.x_df))
def test_calc_A_MRIO(td_small_MRIO): pdt.assert_frame_equal( td_small_MRIO.A, calc_A(td_small_MRIO.Z, td_small_MRIO.x) )
def test_calc_A_arr(td_IO_Data_Miller): npt.assert_array_equal( td_IO_Data_Miller.A_arr, calc_A(td_IO_Data_Miller.Z_arr, td_IO_Data_Miller.x_arr) )
def test_calc_A_df(td_IO_Data_Miller): pdt.assert_frame_equal( td_IO_Data_Miller.A_df, calc_A(td_IO_Data_Miller.Z_df, td_IO_Data_Miller.x_df) )