Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

WIP: More distributions #1413

Merged
merged 17 commits into from Jan 19, 2013
11 changes: 11 additions & 0 deletions doc/src/modules/stats.rst
Expand Up @@ -25,21 +25,32 @@ Continuous Types
.. autofunction:: BetaPrime .. autofunction:: BetaPrime
.. autofunction:: Cauchy .. autofunction:: Cauchy
.. autofunction:: Chi .. autofunction:: Chi
.. autofunction:: ChiNoncentral
.. autofunction:: ChiSquared
.. autofunction:: Dagum .. autofunction:: Dagum
.. autofunction:: Erlang
.. autofunction:: Exponential .. autofunction:: Exponential
.. autofunction:: FDistribution
.. autofunction:: FisherZ
.. autofunction:: Frechet
.. autofunction:: Gamma .. autofunction:: Gamma
.. autofunction:: GammaInverse
.. autofunction:: Kumaraswamy
.. autofunction:: Laplace .. autofunction:: Laplace
.. autofunction:: Logistic .. autofunction:: Logistic
.. autofunction:: LogNormal .. autofunction:: LogNormal
.. autofunction:: Maxwell .. autofunction:: Maxwell
.. autofunction:: Nakagami .. autofunction:: Nakagami
.. autofunction:: Normal .. autofunction:: Normal
.. autofunction:: Pareto .. autofunction:: Pareto
.. autofunction:: QuadraticU
.. autofunction:: RaisedCosine
.. autofunction:: Rayleigh .. autofunction:: Rayleigh
.. autofunction:: StudentT .. autofunction:: StudentT
.. autofunction:: Triangular .. autofunction:: Triangular
.. autofunction:: Uniform .. autofunction:: Uniform
.. autofunction:: UniformSum .. autofunction:: UniformSum
.. autofunction:: VonMises
.. autofunction:: Weibull .. autofunction:: Weibull
.. autofunction:: WignerSemicircle .. autofunction:: WignerSemicircle
.. autofunction:: ContinuousRV .. autofunction:: ContinuousRV
Expand Down
47 changes: 47 additions & 0 deletions sympy/core/tests/test_args.py
Expand Up @@ -696,6 +696,16 @@ def test_sympy__stats__crv_types__ChiPSpace():
assert _test_args(ChiPSpace('X', 1)) assert _test_args(ChiPSpace('X', 1))




def test_sympy__stats__crv_types__ChiNoncentralPSpace():
from sympy.stats.crv_types import ChiNoncentralPSpace
assert _test_args(ChiNoncentralPSpace('X', 1,1))


def test_sympy__stats__crv_types__ChiSquaredPSpace():
from sympy.stats.crv_types import ChiSquaredPSpace
assert _test_args(ChiSquaredPSpace('X', 1))


def test_sympy__stats__crv_types__DagumPSpace(): def test_sympy__stats__crv_types__DagumPSpace():
from sympy.stats.crv_types import DagumPSpace from sympy.stats.crv_types import DagumPSpace
assert _test_args(DagumPSpace('X', 1, 1, 1)) assert _test_args(DagumPSpace('X', 1, 1, 1))
Expand All @@ -706,11 +716,35 @@ def test_sympy__stats__crv_types__ExponentialPSpace():
assert _test_args(ExponentialPSpace('X', 1)) assert _test_args(ExponentialPSpace('X', 1))




def test_sympy__stats__crv_types__FDistributionPSpace():
from sympy.stats.crv_types import FDistributionPSpace
assert _test_args(FDistributionPSpace('X', 1, 1))


def test_sympy__stats__crv_types__FisherZPSpace():
from sympy.stats.crv_types import FisherZPSpace
assert _test_args(FisherZPSpace('X', 1, 1))


def test_sympy__stats__crv_types__FrechetPSpace():
from sympy.stats.crv_types import FrechetPSpace
assert _test_args(FrechetPSpace('X', 1, 1, 1))


def test_sympy__stats__crv_types__GammaInversePSpace():
from sympy.stats.crv_types import GammaInversePSpace
assert _test_args(GammaInversePSpace('X', 1, 1))


def test_sympy__stats__crv_types__GammaPSpace(): def test_sympy__stats__crv_types__GammaPSpace():
from sympy.stats.crv_types import GammaPSpace from sympy.stats.crv_types import GammaPSpace
assert _test_args(GammaPSpace('X', 1, 1)) assert _test_args(GammaPSpace('X', 1, 1))




def test_sympy__stats__crv_types__KumaraswamyPSpace():
from sympy.stats.crv_types import KumaraswamyPSpace
assert _test_args(KumaraswamyPSpace('X', 1, 1))

def test_sympy__stats__crv_types__LaplacePSpace(): def test_sympy__stats__crv_types__LaplacePSpace():
from sympy.stats.crv_types import LaplacePSpace from sympy.stats.crv_types import LaplacePSpace
assert _test_args(LaplacePSpace('X', 0, 1)) assert _test_args(LaplacePSpace('X', 0, 1))
Expand Down Expand Up @@ -746,6 +780,14 @@ def test_sympy__stats__crv_types__ParetoPSpace():
assert _test_args(ParetoPSpace('X', 1, 1)) assert _test_args(ParetoPSpace('X', 1, 1))




def test_sympy__stats__crv_types__QuadraticUPSpace():
from sympy.stats.crv_types import QuadraticUPSpace
assert _test_args(QuadraticUPSpace('X', 1, 2))

def test_sympy__stats__crv_types__RaisedCosinePSpace():
from sympy.stats.crv_types import RaisedCosinePSpace
assert _test_args(RaisedCosinePSpace('X', 1, 1))

def test_sympy__stats__crv_types__RayleighPSpace(): def test_sympy__stats__crv_types__RayleighPSpace():
from sympy.stats.crv_types import RayleighPSpace from sympy.stats.crv_types import RayleighPSpace
assert _test_args(RayleighPSpace('X', 1)) assert _test_args(RayleighPSpace('X', 1))
Expand All @@ -771,6 +813,11 @@ def test_sympy__stats__crv_types__UniformSumPSpace():
assert _test_args(UniformSumPSpace('X', 1)) assert _test_args(UniformSumPSpace('X', 1))




def test_sympy__stats__crv_types__VonMisesPSpace():
from sympy.stats.crv_types import VonMisesPSpace
assert _test_args(VonMisesPSpace('X', 1, 1))


def test_sympy__stats__crv_types__WeibullPSpace(): def test_sympy__stats__crv_types__WeibullPSpace():
from sympy.stats.crv_types import WeibullPSpace from sympy.stats.crv_types import WeibullPSpace
assert _test_args(WeibullPSpace('X', 1, 1)) assert _test_args(WeibullPSpace('X', 1, 1))
Expand Down
11 changes: 7 additions & 4 deletions sympy/stats/__init__.py
Expand Up @@ -53,9 +53,12 @@


import crv_types import crv_types
from crv_types import ( from crv_types import (
Arcsin, Benini, Beta, BetaPrime, Cauchy, Chi, ContinuousRV, Dagum, ContinuousRV,
Exponential, Gamma, Laplace, Logistic, LogNormal, Maxwell, Nakagami, Arcsin, Benini, Beta, BetaPrime, Cauchy, Chi, ChiNoncentral, ChiSquared,
Normal, Pareto, Rayleigh, StudentT, Triangular, Uniform, UniformSum, Dagum, Erlang, Exponential, FDistribution, FisherZ, Frechet, Gamma,
Weibull, WignerSemicircle, GammaInverse, Kumaraswamy, Laplace, Logistic, LogNormal, Maxwell,
Nakagami, Normal, Pareto, QuadraticU, RaisedCosine, Rayleigh,
StudentT, Triangular, Uniform, UniformSum, VonMises, Weibull,
WignerSemicircle
) )
__all__.extend(crv_types.__all__) __all__.extend(crv_types.__all__)