def test_correlations_axis(self, ones_zero): c = _correlation(ones_zero, axis=0, normalize=True) result = np.ones((10, 20)) result[1::2, :] = -1 np.testing.assert_array_equal(c, result) c = _correlation(ones_zero, axis=0, normalize=False) result = np.zeros((10, 20)) result[0::2, :] = 5 np.testing.assert_array_almost_equal(c, result)
def test_correlations_normalization(self, ones_hundred): c = _correlation(ones_hundred, axis=0, normalize=True) result = np.zeros((10, 20)) result[1::2, :] = -0.96078816 result[0::2, :] = 0.96078816 np.testing.assert_array_almost_equal(c, result) c = _correlation(ones_hundred, axis=1, normalize=True) result = np.zeros((10, 20)) np.testing.assert_array_almost_equal(c, result)
def test_correlations_mask(self, ones_hundred): m = np.zeros((10, 20)) m[2:4, :] = 1 c = _correlation(ones_hundred, axis=0, normalize=True, mask=m) print(c) result = np.zeros((10, 20)) result[1::2, :] = -0.96078816 result[0::2, :] = 0.96078816 np.testing.assert_array_almost_equal(c, result)
def test_correlations_wrapping( self, ones_hundred): # Need to do extra checks to assure this is correct m = np.zeros((10, 20)) m[2:4, :] = 1 c = _correlation(ones_hundred, axis=0, normalize=True, wrap=False) print(c) result = np.zeros((10, 20)) result[0::2, :] = 2.26087665 result[1::2, :] = -0.93478899 np.testing.assert_array_almost_equal(c, result)
def test_correlation_ones(self, ones_array): c = _correlation(ones_array) np.testing.assert_array_equal(c, np.zeros((10, 20)))