Python multivariate_normal Examples

Example #1

File: test_random.py Project: tomzhang/mars-1

    def testMultivariateNormal(self):
        mean = [0, 0]
        cov = [[1, 0], [0, 100]]

        t = multivariate_normal(mean, cov, 5000, chunk_size=500)
        self.assertEqual(t.shape, (5000, 2))
        self.assertEqual(t.op.size, (5000, ))

        t = t.tiles()
        self.assertEqual(t.nsplits, ((500, ) * 10, (2, )))
        self.assertEqual(len(t.chunks), 10)
        c = t.chunks[0]
        self.assertEqual(c.shape, (500, 2))
        self.assertEqual(c.op.size, (500, ))

Example #2

File: test_random.py Project: qinxuye/mars

def test_multivariate_normal():
    mean = [0, 0]
    cov = [[1, 0], [0, 100]]

    t = multivariate_normal(mean, cov, 5000, chunk_size=500)
    assert t.shape == (5000, 2)
    assert t.op.size == (5000, )

    t = tile(t)
    assert t.nsplits == ((500, ) * 10, (2, ))
    assert len(t.chunks) == 10
    c = t.chunks[0]
    assert c.shape == (500, 2)
    assert c.op.size == (500, )