| 123456789101112131415161718192021222324252627282930 |
- from sympy.core.basic import Basic
- from sympy.stats.rv import PSpace, _symbol_converter, RandomMatrixSymbol
- class RandomMatrixPSpace(PSpace):
- """
- Represents probability space for
- random matrices. It contains the mechanics
- for handling the API calls for random matrices.
- """
- def __new__(cls, sym, model=None):
- sym = _symbol_converter(sym)
- if model:
- return Basic.__new__(cls, sym, model)
- else:
- return Basic.__new__(cls, sym)
- @property
- def model(self):
- try:
- return self.args[1]
- except IndexError:
- return None
- def compute_density(self, expr, *args):
- rms = expr.atoms(RandomMatrixSymbol)
- if len(rms) > 2 or (not isinstance(expr, RandomMatrixSymbol)):
- raise NotImplementedError("Currently, no algorithm has been "
- "implemented to handle general expressions containing "
- "multiple random matrices.")
- return self.model.density(expr)
|
