def testTransformDf(self):
    df_small = pd.DataFrame(np.array([[1.0, 2.0], [3.0, 4.0]]))
    rotate_mat_np = np.array([[3.0, 4.0], [5.0, 6.0]])
    shift_vec_np = np.array([[(- 1.0)], [(- 2.0)]])
    expected = pd.DataFrame(np.array([[10.0, 15.0], [24.0, 37.0]]))
    df_trans = transform.transform_df(df_small, rotate_mat_np, shift_vec_np)
    pandas_testing.assert_frame_equal(df_trans, expected)
def testTransformDf(self):
    df_small = pd.DataFrame(np.array([[1.0, 2.0], [3.0, 4.0]]))
    rotate_mat_np = np.array([[3.0, 4.0], [5.0, 6.0]])
    shift_vec_np = np.array([[(- 1.0)], [(- 2.0)]])
    expected = pd.DataFrame(np.array([[10.0, 15.0], [24.0, 37.0]]))
    df_trans = transform.transform_df(df_small, rotate_mat_np, shift_vec_np)
    pandas_testing.assert_frame_equal(df_trans, expected)