File: test_TLS.py

package info (click to toggle)
astroml 1.0.2-6
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 932 kB
  • sloc: python: 5,731; makefile: 3
file content (17 lines) | stat: -rw-r--r-- 456 bytes parent folder | download | duplicates (5) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
 
import numpy as np
from numpy.testing import assert_allclose

from ..TLS import TLS_logL


def test_TLS_likelihood_diagonal(rseed=0):
    """Test Total-Least-Squares fit with diagonal covariance"""
    np.random.seed(rseed)

    X = np.random.rand(10, 2)
    dX1 = 0.1 * np.ones((10, 2))
    dX2 = 0.1 * np.array([np.eye(2) for i in range(10)])
    v = np.random.random(2)

    assert_allclose(TLS_logL(v, X, dX1),
                    TLS_logL(v, X, dX2))