def test_calc_corrcoef_ndarray_1(self):
x = np.arange(0, 1, 0.01)
ar_values_sin0 = np.sin(2 * np.pi * x)
ar_values_sin1 = np.sin(2 * np.pi * x + np.pi)
expected = -1.0
actual = calc_corrcoef(ar_values_sin0, ar_values_sin1, delay=0)
self.assertAlmostEqual(expected, actual)
def test_calc_corrcoef_dataframe_2(self):
x = np.arange(0, 1, 0.01)
header0 = ["{}".format(i) for i in range(100)]
header1 = ["{}".format(i + 50) for i in range(100)]
df_values_sin0 = pd.DataFrame(np.sin(2 * np.pi * x), index=header0).T
df_values_sin1 = pd.DataFrame(np.sin(2 * np.pi * x), index=header1)
expected = 1.0
actual = calc_corrcoef(df_values_sin0, df_values_sin1, delay=0)
self.assertAlmostEqual(expected, actual)