def test_full_tensor(self):
     """
     Tests full signal results.
     """
     min_signal = self._mins[0]
     min_diffusivity = self._mins[1]
     full_signals = np.load("signal_full.npy")
     expected_eigs = np.load("expected_full.npy")
     result = tensor_fit(self._ols, self._design, full_signals, min_signal, min_diffusivity)
     result_eigs = np.array([np.sort(arr[0]) for arr in result])
     self.assertTrue(np.allclose(expected_eigs, result_eigs))
    def test_tensor_fit(self):
        """
        tests tensor fit function.
        """

        min_signal = self._mins[0]
        min_diffusivity = self._mins[1]
        result = tensor_fit(self._ols, self._design, self._signals, min_signal, min_diffusivity)
        expected_eigs = np.array([np.sort(arr[0]) for arr in self._expected])
        result_eigs = np.array([np.sort(arr[0]) for arr in result])
        self.assertTrue(np.allclose(expected_eigs, result_eigs))