def test_process_column_lr(self):
        l_dist_matrix = self.dist_matrix.copy()
        r_dist_matrix = self.dist_matrix.copy()

        mmp = MultidimensionalMatrixProfileLR()
        self.mock_initialise(mmp)

        for diag in range(-self.dist_matrix.shape[1] + 1,
                          self.dist_matrix.shape[2]):
            diag_ind = diag_indices_of(self.dist_matrix[0], diag)
            if diag < 0:
                l_dist_matrix[:, diag_ind[0], diag_ind[1]] = np.inf
            else:
                r_dist_matrix[:, diag_ind[0], diag_ind[1]] = np.inf

        for column in range(self.dist_matrix.shape[2]):
            mmp.process_column(column, self.dist_matrix[:, :, column])

        correct_lmp, correct_lindex, correct_ldims = self.bruteforce_mmp(
            l_dist_matrix)
        correct_rmp, correct_rindex, correct_rdims = self.bruteforce_mmp(
            r_dist_matrix)

        npt.assert_allclose(mmp.md_matrix_profile_left, correct_lmp)
        npt.assert_allclose(mmp.md_matrix_profile_right, correct_rmp)
        npt.assert_equal(mmp.md_profile_index_left, correct_lindex)
        npt.assert_equal(mmp.md_profile_index_right, correct_rindex)
        for i, dims in enumerate(correct_ldims):
            sorted_mmp_ldims = np.sort(mmp.md_profile_dimension_left[i],
                                       axis=0)
            sorted_mmp_rdims = np.sort(mmp.md_profile_dimension_right[i],
                                       axis=0)
            npt.assert_equal(sorted_mmp_ldims, correct_ldims[i])
            npt.assert_equal(sorted_mmp_rdims, correct_rdims[i])
    def test_process_column(self):
        mmp = MultidimensionalMatrixProfileLR()
        self.mock_initialise(mmp)

        for column in range(0, self.dist_matrix.shape[2]):
            mmp.process_column(column, self.dist_matrix[:, :, column])

        correct_mp, correct_index, correct_dims = self.bruteforce_mmp(
            self.dist_matrix)

        npt.assert_allclose(mmp.md_matrix_profile(), correct_mp)
        npt.assert_equal(mmp.md_profile_index(), correct_index)
        for i, dims in enumerate(correct_dims):
            sorted_mmp_dims = np.sort(mmp.md_profile_dimensions()[i], axis=0)
            npt.assert_equal(sorted_mmp_dims, dims)
    def test_process_column_partial_calculation(self):
        part_dist_matrix = np.full_like(self.dist_matrix, np.inf)

        mmp = MultidimensionalMatrixProfileLR()
        self.mock_initialise(mmp)

        for column in [0, 3, 4, 6, 9, 12, 13]:
            mmp.process_column(column, self.dist_matrix[:, :, column])
            part_dist_matrix[:, :, column] = self.dist_matrix[:, :, column]

        correct_mp, correct_index, correct_dims = self.bruteforce_mmp(
            part_dist_matrix)

        npt.assert_allclose(mmp.md_matrix_profile(), correct_mp)
        npt.assert_equal(mmp.md_profile_index(), correct_index)
        for i, dims in enumerate(correct_dims):
            sorted_mmp_dims = np.sort(mmp.md_profile_dimensions()[i], axis=0)
            npt.assert_equal(sorted_mmp_dims, dims)
    def test_streaming_process_column(self):
        mmp = MultidimensionalMatrixProfileLR()
        mmp.initialise(3, 5, 7)

        available_data = np.full((3, 5, 7), np.inf)
        available_data[:, 0:5, 0] = self.dist_matrix[:, 0:5, 0]
        available_data[:, 0:5, 4] = self.dist_matrix[:, 0:5, 4]
        available_data[:, 0:5, 5] = self.dist_matrix[:, 0:5, 5]
        mmp.process_column(0, self.dist_matrix[:, 0:5, 0])
        mmp.process_column(4, self.dist_matrix[:, 0:5, 4])
        mmp.process_column(5, self.dist_matrix[:, 0:5, 5])

        correct_mp, correct_index, correct_dims = self.bruteforce_mmp(
            available_data)
        npt.assert_allclose(mmp.md_matrix_profile(), correct_mp)
        npt.assert_equal(mmp.md_profile_index(), correct_index)
        for i, dims in enumerate(correct_dims):
            sorted_mmp_dims = np.sort(mmp.md_profile_dimensions()[i], axis=0)
            npt.assert_equal(sorted_mmp_dims, dims)

        mmp.shift_series(2)
        mmp.shift_query(3)
        available_data = np.full((3, 8, 9), np.inf)
        available_data[:, 0:5, 4] = self.dist_matrix[:, 0:5, 4]
        available_data[:, 0:5, 5] = self.dist_matrix[:, 0:5, 5]

        correct_mp, correct_index, correct_dims = self.bruteforce_mmp(
            available_data)
        npt.assert_allclose(mmp.md_matrix_profile(), correct_mp[:, 2:])
        npt.assert_equal(mmp.md_profile_index(), correct_index[:, 2:])
        for i, dims in enumerate(correct_dims):
            sorted_mmp_dims = np.sort(mmp.md_profile_dimensions()[i], axis=0)
            npt.assert_equal(sorted_mmp_dims, dims[:, 2:])

        mmp.process_column(0, self.dist_matrix[:, 3:8, 2])
        mmp.process_column(2, self.dist_matrix[:, 3:8, 4])
        mmp.process_column(3, self.dist_matrix[:, 3:8, 5])
        mmp.process_column(5, self.dist_matrix[:, 3:8, 7])
        available_data[:, 3:8, 2] = self.dist_matrix[:, 3:8, 2]
        available_data[:, 3:8, 4] = self.dist_matrix[:, 3:8, 4]
        available_data[:, 3:8, 5] = self.dist_matrix[:, 3:8, 5]
        available_data[:, 3:8, 7] = self.dist_matrix[:, 3:8, 7]

        correct_mp, correct_index, correct_dims = self.bruteforce_mmp(
            available_data)
        npt.assert_allclose(mmp.md_matrix_profile(), correct_mp[:, 2:])
        npt.assert_equal(mmp.md_profile_index(), correct_index[:, 2:])
        for i, dims in enumerate(correct_dims):
            sorted_mmp_dims = np.sort(mmp.md_profile_dimensions()[i], axis=0)
            npt.assert_equal(sorted_mmp_dims, dims[:, 2:])