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))
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))
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))
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) )
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) )