Пример #1
0
    def test_init(self):
        """Test __init__."""

        skst = MvSN(ndim=3)

        self.assertIsInstance(skst.lam, np.ndarray)

        lam = [.5, 1.5]
        skst = MvSN(ndim=len(lam), lam=lam)

        npt.assert_array_equal(skst.lam, np.array(lam))

        mu, sigma = [.5, .4], np.ones((2, 2))
        skst = MvSN(ndim=len(lam), lam=lam, mu=mu, sigma=sigma)

        npt.assert_array_equal(skst.mu, np.array(mu))
        npt.assert_array_equal(skst.sigma, np.array(sigma))
        npt.assert_array_equal(skst.const_mu(), np.array(mu))
        npt.assert_array_equal(skst.const_sigma(), np.array(sigma))

        lam = [1.5, .5]
        skst.from_theta(np.array(lam))

        npt.assert_array_equal(skst.lam, np.array(lam))

        size = len(lam)
        data = np.random.normal(size=size)
        skst = MvSN(ndim=len(lam), data=data)

        npt.assert_array_equal(skst.data, np.atleast_2d(data))
Пример #2
0
    def test_init(self):
        """Test __init__."""

        skst = MvSN(ndim=3)

        self.assertIsInstance(skst.lam, np.ndarray)

        lam = [.5, 1.5]
        skst = MvSN(ndim=len(lam), lam=lam)

        npt.assert_array_equal(skst.lam, np.array(lam))

        mu, sigma = [.5, .4], np.ones((2, 2))
        skst = MvSN(ndim=len(lam), lam=lam, mu=mu, sigma=sigma)

        npt.assert_array_equal(skst.mu, np.array(mu))
        npt.assert_array_equal(skst.sigma, np.array(sigma))
        npt.assert_array_equal(skst.const_mu(), np.array(mu))
        npt.assert_array_equal(skst.const_sigma(), np.array(sigma))

        lam = [1.5, .5]
        skst.from_theta(np.array(lam))

        npt.assert_array_equal(skst.lam, np.array(lam))

        size = len(lam)
        data = np.random.normal(size=size)
        skst = MvSN(ndim=len(lam), data=data)

        npt.assert_array_equal(skst.data, np.atleast_2d(data))