Skip to content
Snippets Groups Projects
Verified Commit b0b09a24 authored by Anton Akhmerov's avatar Anton Akhmerov
Browse files

improve sympy-ness check; closes #425

parent 58f3620a
No related branches found
No related tags found
1 merge request!409Sympy compat
......@@ -16,16 +16,13 @@ import sympy.abc
import sympy.physics.quantum
from sympy.core.function import AppliedUndef
from sympy.core.sympify import converter
from sympy.core.core import all_classes as sympy_classes
from sympy.core import Basic
from sympy.physics.matrices import msigma as _msigma
import warnings
from .._common import reraise_warnings
# TODO: remove when sympy correctly includes MutableDenseMatrix (lol).
sympy_classes = set(sympy_classes) | {sympy.MutableDenseMatrix}
momentum_operators = sympy.symbols('k_x k_y k_z', commutative=False)
position_operators = sympy.symbols('x y z', commutative=False)
......@@ -152,7 +149,7 @@ def sympify(expr, locals=None):
stored_value = None
# if ``expr`` is already a ``sympy`` object we may terminate a code path
if isinstance(expr, tuple(sympy_classes)):
if isinstance(expr, Basic):
if locals:
warnings.warn('Input expression is already SymPy object: '
'"locals" will not be used.',
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment